lunny/keras
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add support for custom padding value in sequence

Browse Source
This commit is contained in:
fchollet 2015-06-26 17:44:56 -07:00
parent ea0b7d263c
commit 9c8e0d43f3
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
keras/preprocessing/sequence.py
View File

@ -4,7 +4,7 @@ import numpy as np
import random
from six.moves import range
def pad_sequences(sequences, maxlen=None, dtype='int32', padding='pre'):
def pad_sequences(sequences, maxlen=None, dtype='int32', padding='pre', value=0.):
"""
Pad each sequence to the same length:
the length of the longuest sequence.
@ -20,7 +20,7 @@ def pad_sequences(sequences, maxlen=None, dtype='int32', padding='pre'):
if maxlen is None:
maxlen = np.max(lengths)
x = np.zeros((nb_samples, maxlen)).astype(dtype)
x = (np.ones((nb_samples, maxlen)) * value).astype(dtype)
for idx, s in enumerate(sequences):
if padding == 'post':
x[idx, :lengths[idx]] = s[:maxlen]