diff options
Diffstat (limited to 'notebooks/cc/reference_annotations/hyperparameter_search.py')
| -rw-r--r-- | notebooks/cc/reference_annotations/hyperparameter_search.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/notebooks/cc/reference_annotations/hyperparameter_search.py b/notebooks/cc/reference_annotations/hyperparameter_search.py index a7286aa0..fc42d148 100644 --- a/notebooks/cc/reference_annotations/hyperparameter_search.py +++ b/notebooks/cc/reference_annotations/hyperparameter_search.py @@ -14,11 +14,11 @@ import optuna import pandas as pd import torch from rapidfuzz import fuzz +from sec_certs.model.references.annotator_trainer import ReferenceAnnotatorTrainer +from sec_certs.model.references.segment_extractor import ReferenceSegmentExtractor from sklearn.metrics import f1_score from sec_certs.dataset import CCDataset -from sec_certs.model.references.annotator_trainer import ReferenceAnnotatorTrainer -from sec_certs.model.references.segment_extractor import ReferenceSegmentExtractor from sec_certs.utils.helpers import compute_heuristics_version from sec_certs.utils.nlp import prec_recall_metric |