diff --git a/keras/datasets/imdb.py b/keras/datasets/imdb.py index 9697d4033..64e3be1f5 100644 --- a/keras/datasets/imdb.py +++ b/keras/datasets/imdb.py @@ -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 diff --git a/keras/datasets/reuters.py b/keras/datasets/reuters.py index f94d55a58..399a130a6 100644 --- a/keras/datasets/reuters.py +++ b/keras/datasets/reuters.py @@ -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