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

minor...

parent 80289922
Branches
No related tags found
No related merge requests found
...@@ -507,7 +507,7 @@ class IcingIntensityNN: ...@@ -507,7 +507,7 @@ class IcingIntensityNN:
print('num test samples: ', tst_idxs.shape[0]) print('num test samples: ', tst_idxs.shape[0])
print('setup_pipeline: Done') print('setup_pipeline: Done')
def setup_test_pipeline(self, filename_l1b, filename_l2, seed=None, shuffle=False): def setup_test_pipeline(self, filename_l1b, filename_l2):
if filename_l1b is not None: if filename_l1b is not None:
self.h5f_l1b_tst = h5py.File(filename_l1b, 'r') self.h5f_l1b_tst = h5py.File(filename_l1b, 'r')
...@@ -521,10 +521,6 @@ class IcingIntensityNN: ...@@ -521,10 +521,6 @@ class IcingIntensityNN:
time = h5f['time'] time = h5f['time']
tst_idxs = np.arange(time.shape[0]) tst_idxs = np.arange(time.shape[0])
self.num_data_samples = len(tst_idxs) self.num_data_samples = len(tst_idxs)
if seed is not None:
np.random.seed(seed)
if shuffle:
np.random.shuffle(tst_idxs)
self.get_test_dataset(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 to comment