diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py
index 985e7032106bb1821146b576ff42c613970b0054..aee3e3b536ad443acea973b076d615c4161dc592 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)