From bcbfcc000ce07078786bae5f66f8b30dbe82fb64 Mon Sep 17 00:00:00 2001 From: Andrew Poliakov Date: Wed, 26 Apr 2017 05:48:42 +0300 Subject: [PATCH] Fix oov_char=None case in IMDB/Reuters datasets (#6397) Closes: #3688 --- keras/datasets/imdb.py | 2 +- keras/datasets/reuters.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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