Add files via upload

mineraldragon · Jan 2, 2022 · 2298ca5 · 2298ca5
1 parent c9dce74
commit 2298ca5
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/variables.py b/variables.py
@@ -27,16 +27,16 @@ class Vars:
     RECORDINGS_DIR = 'recordings'
     RESULTS_DIR = 'results'
     MODELS_DIR = 'models'
-    CONFIDENCE_THRESHOLD = 0.95
-    TRAINING_BATCH_SIZE = 16
-    TRAINING_EPOCHS = 15
+    CONFIDENCE_THRESHOLD = 0.9
+    TRAINING_BATCH_SIZE = 128
+    TRAINING_EPOCHS = 8
     DETECTION_LENGTH_RATIO = 0.5
     WINDOW_LENGTHS = {'Chi': 0.25,'Tr': 0.25,'Ph': 0.40,'Tw': 0.5}
     SEGMENT_LENGTH = 0.45
     SEGMENT_STEP = 0.04
     VALIDATION_RATIO = 0.2
     TEST_RATIO = 0.1
-    TRAINING_SEGMENTS_PER_CALL = 35000
+    TRAINING_SEGMENTS_PER_CALL = 20000
     TESTING_SEGMENTS_PER_CALL = int(round(TRAINING_SEGMENTS_PER_CALL * TEST_RATIO))
     VALIDATION_SEGMENTS_PER_CALL = int(round(TRAINING_SEGMENTS_PER_CALL * VALIDATION_RATIO))
     TDATA_FILENAME = 'acdc.tdata'