From d333db1f0922202667508bf3e74f07815ce6a1a1 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Mon, 12 Jul 2021 19:18:17 -0500
Subject: [PATCH] minor...
---
modules/deeplearning/icing_cnn.py | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py
index 0fcde076..81e6b716 100644
--- a/modules/deeplearning/icing_cnn.py
+++ b/modules/deeplearning/icing_cnn.py
@@ -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)
--
GitLab