demo added

Ze-sys · Sep 5, 2022 · 8ca2014 · 8ca2014
1 parent 585580b
commit 8ca2014
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -16,6 +16,11 @@ In addition to the hydrophone spectrogram images used in the paper, the models i
 
 Details to come soon.
 
+#### The model training component:
+![model-trainer-demo](model-trainer-demo.gif)
+
+#### The model testing component:
+![model-tester-demo](model-tester-demo.gif)
 
 #### Credits: 
 

diff --git a/model-tester-demo.gif b/model-tester-demo.gif
diff --git a/model-trainer-demo.gif b/model-trainer-demo.gif
diff --git a/pages/modelBuilder.py b/pages/modelBuilder.py
@@ -770,7 +770,7 @@ def show_loss_and_accuracy_curves(history_):
     print_stats(preds, test_labels, 'Anomalous (test) set reconstruction metrics:', tag='anomalous')
 
     print(logdir)
-    write_current_model_name(threshold=threshold[0].round(6))  # to csv file
+    write_current_model_name()  # to csv file
 
 
     assert autoencoder is not None, "Current model is empty. Run the model to save the the weights."