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

Fix oov_char=None case in IMDB/Reuters datasets (#6397)

Browse Source 
Closes: #3688
This commit is contained in:
Andrew Poliakov 2017-04-26 05:48:42 +03:00 committed by François Chollet
parent 54a417f616
commit bcbfcc000c
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
keras/datasets/imdb.py
View File

@ -100,7 +100,7 @@ def load_data(path='imdb.npz', num_words=None, skip_top=0,
for x in xs:
nx = []
for w in x:
if w >= num_words or w < skip_top:
if skip_top <= w < num_words:
nx.append(w)
new_xs.append(nx)
xs = new_xs

2
keras/datasets/reuters.py
View File

@ -84,7 +84,7 @@ def load_data(path='reuters.npz', num_words=None, skip_top=0,
for x in xs:
nx = []
for w in x:
if w >= num_words or w < skip_top:
if skip_top <= w < num_words:
nx.append(w)
new_xs.append(nx)
xs = new_xs