From 80a62a6e32860e32380dfcdd9c460a0be64dca8c Mon Sep 17 00:00:00 2001 From: Harrison Date: Thu, 1 Jun 2023 16:16:13 -0500 Subject: [PATCH] Restructured in preparation for packaging --- trainers.py => src/mltraining/trainers.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) rename trainers.py => src/mltraining/trainers.py (98%) diff --git a/trainers.py b/src/mltraining/trainers.py similarity index 98% rename from trainers.py rename to src/mltraining/trainers.py index 06e47ba..cc3223b 100644 --- a/trainers.py +++ b/src/mltraining/trainers.py @@ -57,7 +57,7 @@ class TokenClassificationTrainer: ) labels = [] - for i, label in enumerate(examples[f"ner_tags"]): + for i, label in enumerate(examples["ner_tags"]): word_ids = tokenized_inputs.word_ids( batch_index=i ) # Map tokens to their respective word. @@ -100,7 +100,7 @@ class TokenClassificationTrainer: results: dict[str, float] = self._evaluator.compute( predictions=true_predictions, references=true_labels ) # type: ignore - + return { "precision": results["overall_precision"], "recall": results["overall_recall"],