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
Show whitespace changes
Inline Side-by-side
modules/deeplearning/icing_cnn.py
+ 2
− 2
View file @ 7fb9acb8
...@@ -924,8 +924,8 @@ class IcingIntensityNN: ...@@ -924,8 +924,8 @@ class IcingIntensityNN:
self.test_loss.reset_states() self.test_loss.reset_states()
self.test_accuracy.reset_states() self.test_accuracy.reset_states()
for data0, label in self.test_dataset: for data0, data1, label in self.test_dataset:
ds = tf.data.Dataset.from_tensor_slices((data0, label)) ds = tf.data.Dataset.from_tensor_slices((data0, data1, label))
ds = ds.batch(BATCH_SIZE) ds = ds.batch(BATCH_SIZE)
for mini_batch_test in ds: for mini_batch_test in ds:
self.predict(mini_batch_test) self.predict(mini_batch_test)
... ...
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment