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

snapshot...

parent 39e6f313
Branches
No related tags found
No related merge requests found
...@@ -535,14 +535,19 @@ class IcingIntensityNN: ...@@ -535,14 +535,19 @@ class IcingIntensityNN:
self.test_loss.reset_states() self.test_loss.reset_states()
self.test_accuracy.reset_states() self.test_accuracy.reset_states()
self.test_auc.reset_states()
self.test_recall.reset_states()
self.test_precision.reset_states()
for data0, label in self.test_dataset: for data0, label in self.test_dataset:
ds = tf.data.Dataset.from_tensor_slices((data0, label)) ds = tf.data.Dataset.from_tensor_slices((data0, label))
ds = ds.batch(BATCH_SIZE) ds = ds.batch(BATCH_SIZE)
for mini_batch in ds: for mini_batch in ds:
self.test_step(mini_batch) self.test_step(mini_batch)
print('loss, acc : ', self.test_loss.result().numpy(), self.test_accuracy.result().numpy()) print('loss, acc, auc, recall, precision: ', self.test_loss.result().numpy(), self.test_accuracy.result().numpy(),
print('---------------------------------------------------------') self.test_auc.result().numpy(), self.test_recall.result().numpy(), self.test_precision.result().numpy())
print('--------------------------------------------------')
ckpt_manager.save() ckpt_manager.save()
if self.DISK_CACHE and epoch == 0: if self.DISK_CACHE and epoch == 0:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment