experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Prevent Keras dataset loading from affecting the global RNG

Browse Source 
PiperOrigin-RevId: 316533425
Change-Id: I6099847f9a7ead24786fb2fecd5ba488f53456e6
This commit is contained in:
Francois Chollet 2020-06-15 13:30:30 -07:00 committed by TensorFlower Gardener
parent df4ea0c1a5
commit 52736a6adc
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/python/keras/datasets/imdb.py
View File

@ -113,14 +113,14 @@ def load_data(path='imdb.npz',
x_train, labels_train = f['x_train'], f['y_train']
x_test, labels_test = f['x_test'], f['y_test']
np.random.seed(seed)
rng = np.random.RandomState(seed)
indices = np.arange(len(x_train))
np.random.shuffle(indices)
rng.shuffle(indices)
x_train = x_train[indices]
labels_train = labels_train[indices]
indices = np.arange(len(x_test))
np.random.shuffle(indices)
rng.shuffle(indices)
x_test = x_test[indices]
labels_test = labels_test[indices]