Skip to content
Snippets Groups Projects
Commit 54076686 authored by Eva Bardou's avatar Eva Bardou :frog: Committed by Yoann Schneider
Browse files

Fix typo + Nits

parent c844e024
No related branches found
No related tags found
1 merge request!39Utils method to print a Markdown table
Pipeline #150450 passed
......@@ -14,7 +14,7 @@ from nerval.parse import (
look_for_further_entity_part,
parse_bio,
)
from nerval.utils import print_mardown_table, print_result_compact, print_results
from nerval.utils import print_markdown_table, print_result_compact, print_results
logger = logging.getLogger(__name__)
......@@ -395,4 +395,4 @@ def run_multiple(file_csv: Path, folder: Path, threshold: int, verbose: bool):
round(recall / count, 3),
round(f1 / count, 3),
]
print_mardown_table(["Precision", "Recall", "F1"], [result])
print_markdown_table(["Precision", "Recall", "F1"], [result])
from prettytable import MARKDOWN, PrettyTable
def print_mardown_table(header: list[str], rows: list[list]):
def print_markdown_table(header: list[str], rows: list[list]) -> None:
"""Prints a Markdown table filled with the provided header and rows."""
table = PrettyTable()
table.field_names = header
table.set_style(MARKDOWN)
......@@ -9,7 +10,7 @@ def print_mardown_table(header: list[str], rows: list[list]):
print(table)
def print_results(scores: dict):
def print_results(scores: dict) -> None:
"""Display final results.
None values are kept to indicate the absence of a certain tag in either annotation or prediction.
......@@ -32,13 +33,13 @@ def print_results(scores: dict):
],
)
print_mardown_table(
print_markdown_table(
["tag", "predicted", "matched", "Precision", "Recall", "F1", "Support"],
results,
)
def print_result_compact(scores: dict):
def print_result_compact(scores: dict) -> None:
result = [
"All",
scores["All"]["predicted"],
......@@ -48,7 +49,7 @@ def print_result_compact(scores: dict):
round(scores["All"]["F1"], 3),
scores["All"]["Support"],
]
print_mardown_table(
print_markdown_table(
["tag", "predicted", "matched", "Precision", "Recall", "F1", "Support"],
[result],
)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment