From 7fb9acb82b0da13b8e4bf570f1eda0b4f91d5e0e Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Fri, 22 Oct 2021 16:11:51 -0500
Subject: [PATCH] bug fix
---
modules/deeplearning/icing_cnn.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py
index 78e551d7..2fd21c81 100644
--- a/modules/deeplearning/icing_cnn.py
+++ b/modules/deeplearning/icing_cnn.py
@@ -924,8 +924,8 @@ class IcingIntensityNN:
self.test_loss.reset_states()
self.test_accuracy.reset_states()
- for data0, label in self.test_dataset:
- ds = tf.data.Dataset.from_tensor_slices((data0, label))
+ for data0, data1, label in self.test_dataset:
+ ds = tf.data.Dataset.from_tensor_slices((data0, data1, label))
ds = ds.batch(BATCH_SIZE)
for mini_batch_test in ds:
self.predict(mini_batch_test)
--
GitLab