hardcode optimal hyperparams for embeddings

crocs-muni · Nov 23, 2023 · 14df232 · 14df232
1 parent 68b1c2d
commit 14df232
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/sec_certs/model/references_nlp/feature_extraction.py b/src/sec_certs/model/references_nlp/feature_extraction.py
@@ -155,11 +155,11 @@ def _build_transformer_embeddings(
             segments,
             prec_recall_metric,
             mode=mode,
-            n_iterations=34,
-            n_epochs=1,
-            learning_rate=0.01,
+            n_iterations=8,
+            n_epochs=2,
+            learning_rate=1.23e-5,
             batch_size=16,
-            segmenter_metric="f1",
+            segmenter_metric="balanced_accuracy",
             ensemble_soft_voting_power=2,
             show_progress_bar=False,
         )