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

bug fix

parent 48c567e2
Branches
No related tags found
No related merge requests found
......@@ -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)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment