Skip to content
Snippets Groups Projects

Evaluate predictions with nerval

Merged Manon Blanco requested to merge nerval-evaluate into main
3 files
+ 10
7
Compare changes
  • Side-by-side
  • Inline
Files
3
+ 1
3
@@ -110,8 +110,6 @@ def eval(rank, config: dict, nerval_threshold: float, mlflow_logging: bool):
if "ner" not in metric_names:
return
print()
def inferences_to_parsed_bio(attr: str):
bio_values = []
for inference in inferences:
@@ -133,7 +131,7 @@ def eval(rank, config: dict, nerval_threshold: float, mlflow_logging: bool):
continue
scores = evaluate(ground_truths, predictions, nerval_threshold)
print(set_name)
print(f"\n#### {set_name}\n")
print_results(scores)
Loading