Skip to content
Snippets Groups Projects
Commit d333db1f authored by tomrink's avatar tomrink
Browse files

minor...

parent 0720db0c
No related merge requests found
......@@ -308,7 +308,7 @@ class IcingIntensityNN:
print('num test samples: ', tst_idxs.shape[0])
print('setup_pipeline: Done')
def setup_test_pipeline(self, filename, seed=None):
def setup_test_pipeline(self, filename, seed=None, shuffle=False):
self.filename_tst = filename
self.h5f_tst = h5py.File(filename, 'r')
......@@ -317,7 +317,8 @@ class IcingIntensityNN:
self.num_data_samples = len(tst_idxs)
if seed is not None:
np.random.seed(seed)
np.random.shuffle(tst_idxs)
if shuffle:
np.random.shuffle(tst_idxs)
self.get_test_dataset(tst_idxs)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment