diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/sec_certs/model/references_nlp/feature_extraction.py | 8 |
1 files changed, 4 insertions, 4 deletions
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, ) |