diff --git a/run_snomed_german_recommender.py b/run_snomed_german_recommender.py index 40bd0f6..0fc3b06 100644 --- a/run_snomed_german_recommender.py +++ b/run_snomed_german_recommender.py @@ -14,12 +14,6 @@ from utils import handle_dates -from transformers import logging as tf_logging -import logging - -logging.basicConfig(level=logging.INFO) -tf_logging.set_verbosity_info() - class xMENSNOMEDLinker(Classifier): def __init__(self, linker: EntityLinker, top_k = 3): self.linker = linker @@ -36,8 +30,7 @@ def predict(self, cas: Cas, layer: str, feature: str, project_id: str, document_ # TODO: support re-ranking (see GraSCCo_Evaluation.ipynb) for anno, pred in zip(annos, preds): - for concept in pred['normalized'][0:self.top_k]: - print(concept) + for concept in pred['normalized'][0:self.top_k]: sctid = concept['db_id'] score = concept['score'] prediction = create_span_prediction(cas, layer, feature, anno.begin, anno.end, f"http://snomed.info/id/{sctid}", score=score)