From 0e93a89107366117ffa616c10a73ef42fea12685 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Tue, 18 May 2021 21:47:30 -0500
Subject: [PATCH] minor...

---
 modules/deeplearning/icing_cnn.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py
index 985e7032..aee3e3b5 100644
--- a/modules/deeplearning/icing_cnn.py
+++ b/modules/deeplearning/icing_cnn.py
@@ -306,6 +306,7 @@ class IcingIntensityNN:
 
         time = self.h5f_tst['time']
         tst_idxs = np.arange(time.shape[0])
+        self.num_data_samples = len(tst_idxs)
         if seed is not None:
             np.random.seed(seed)
         np.random.shuffle(tst_idxs)
-- 
GitLab