From 8d1289ed490575e31f09f5942d624390f632fde0 Mon Sep 17 00:00:00 2001
From: MasterPhooey <106418429+MasterPhooey@users.noreply.github.com>
Date: Sun, 5 Jan 2025 16:47:10 -0600
Subject: [PATCH] Update advanced_training_notebook.ipynb

---
 advanced_training_notebook.ipynb | 24 ++++++++++++------------
 1 file changed, 12 insertions(+), 12 deletions(-)

diff --git a/advanced_training_notebook.ipynb b/advanced_training_notebook.ipynb
index f598fda..30f9ce3 100644
--- a/advanced_training_notebook.ipynb
+++ b/advanced_training_notebook.ipynb
@@ -527,7 +527,7 @@
     "config[\"features\"] = [\n",
     "    {\n",
     "        \"features_dir\": \"generated_augmented_features\",\n",
-    "        \"sampling_weight\": 5.0,  # Increased\n",
+    "        \"sampling_weight\": 2.0,  # Increased\n",
     "        \"penalty_weight\": 1.0,\n",
     "        \"truth\": True,\n",
     "        \"truncation_strategy\": \"truncate_start\",\n",
@@ -535,7 +535,7 @@
     "    },\n",
     "    {\n",
     "        \"features_dir\": \"negative_datasets/speech\",\n",
-    "        \"sampling_weight\": 8.0,  # Adjusted\n",
+    "        \"sampling_weight\": 10.0,  # Adjusted\n",
     "        \"penalty_weight\": 1.0,\n",
     "        \"truth\": False,\n",
     "        \"truncation_strategy\": \"random\",\n",
@@ -543,7 +543,7 @@
     "    },\n",
     "    {\n",
     "        \"features_dir\": \"negative_datasets/dinner_party\",\n",
-    "        \"sampling_weight\": 8.0,  # Adjusted\n",
+    "        \"sampling_weight\": 10.0,  # Adjusted\n",
     "        \"penalty_weight\": 1.0,\n",
     "        \"truth\": False,\n",
     "        \"truncation_strategy\": \"random\",\n",
@@ -569,20 +569,20 @@
     "\n",
     "config[\"training_steps\"] = [20000]  # Increased\n",
     "config[\"positive_class_weight\"] = [1]\n",
-    "config[\"negative_class_weight\"] = [15]  # Adjusted\n",
-    "config[\"learning_rates\"] = [0.0005]  # Adjusted\n",
+    "config[\"negative_class_weight\"] = [20]  # Adjusted\n",
+    "config[\"learning_rates\"] = [0.001]  # Adjusted\n",
     "config[\"batch_size\"] = 128\n",
     "\n",
-    "config[\"time_mask_max_size\"] = [30]  # Enabled SpecAugment\n",
-    "config[\"time_mask_count\"] = [2]\n",
-    "config[\"freq_mask_max_size\"] = [15]\n",
-    "config[\"freq_mask_count\"] = [2]\n",
+    "config[\"time_mask_max_size\"] = [0]  # Enabled SpecAugment\n",
+    "config[\"time_mask_count\"] = [0]\n",
+    "config[\"freq_mask_max_size\"] = [0]\n",
+    "config[\"freq_mask_count\"] = [0]\n",
     "\n",
-    "config[\"eval_step_interval\"] = 1000  # Adjusted\n",
-    "config[\"clip_duration_ms\"] = 2000  # Increased\n",
+    "config[\"eval_step_interval\"] = 500  # Adjusted\n",
+    "config[\"clip_duration_ms\"] = 1500  # Increased\n",
     "\n",
     "config[\"target_minimization\"] = 0.9\n",
-    "config[\"minimization_metric\"] = \"false_positive_rate\"  # Updated\n",
+    "config[\"minimization_metric\"] = None  # Updated\n",
     "config[\"maximization_metric\"] = \"average_viable_recall\"\n",
     "\n",
     "with open(os.path.join(\"training_parameters.yaml\"), \"w\") as file:\n",