From 14df232736e43dde9de47d3bbe506bdaae6fa53d Mon Sep 17 00:00:00 2001
From: adamjanovsky
Date: Thu, 23 Nov 2023 09:23:11 +0100
Subject: hardcode optimal hyperparams for embeddings

---
 src/sec_certs/model/references_nlp/feature_extraction.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

(limited to 'src')

diff --git a/src/sec_certs/model/references_nlp/feature_extraction.py b/src/sec_certs/model/references_nlp/feature_extraction.py
index eed90959..c875cfab 100644
--- 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,
         )
-- 
cgit v1.3.1