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

Prevent Keras dataset loading from affecting the global RNG seed.

Browse Source 
PiperOrigin-RevId: 316527944
Change-Id: I13fc997ffafc02f25b94e45265c7aa97b6efc6c4
This commit is contained in:
Francois Chollet 2020-06-15 13:02:46 -07:00 committed by TensorFlower Gardener
parent 540852285d
commit 18e0e6450b
2 changed files with 4 additions and 4 deletions
tensorflow/python/keras/datasets
boston_housing.pyreuters.py

4
tensorflow/python/keras/datasets/boston_housing.py
View File

@ -67,9 +67,9 @@ def load_data(path='boston_housing.npz', test_split=0.2, seed=113):
x = f['x']
y = f['y']
np.random.seed(seed)
rng = np.random.RandomState(seed)
indices = np.arange(len(x))
np.random.shuffle(indices)
rng.shuffle(indices)
x = x[indices]
y = y[indices]

4
tensorflow/python/keras/datasets/reuters.py
View File

@ -119,9 +119,9 @@ def load_data(path='reuters.npz',
with np.load(path, allow_pickle=True) as f:
xs, labels = f['x'], f['y']
np.random.seed(seed)
rng = np.random.RandomState(seed)
indices = np.arange(len(xs))
np.random.shuffle(indices)
rng.shuffle(indices)
xs = xs[indices]
labels = labels[indices]