parent
54a417f616
commit
bcbfcc000c
@ -100,7 +100,7 @@ def load_data(path='imdb.npz', num_words=None, skip_top=0,
|
|||||||
for x in xs:
|
for x in xs:
|
||||||
nx = []
|
nx = []
|
||||||
for w in x:
|
for w in x:
|
||||||
if w >= num_words or w < skip_top:
|
if skip_top <= w < num_words:
|
||||||
nx.append(w)
|
nx.append(w)
|
||||||
new_xs.append(nx)
|
new_xs.append(nx)
|
||||||
xs = new_xs
|
xs = new_xs
|
||||||
|
@ -84,7 +84,7 @@ def load_data(path='reuters.npz', num_words=None, skip_top=0,
|
|||||||
for x in xs:
|
for x in xs:
|
||||||
nx = []
|
nx = []
|
||||||
for w in x:
|
for w in x:
|
||||||
if w >= num_words or w < skip_top:
|
if skip_top <= w < num_words:
|
||||||
nx.append(w)
|
nx.append(w)
|
||||||
new_xs.append(nx)
|
new_xs.append(nx)
|
||||||
xs = new_xs
|
xs = new_xs
|
||||||
|
Loading…
Reference in New Issue
Block a user