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

minor...

parent 0720db0c
No related branches found
No related tags found
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.
Finish editing this message first!
Please register or to comment