diff --git a/modules/bert/finetuning-legal-bert-classifier.ipynb b/modules/bert/finetuning-legal-bert-classifier.ipynb index c1040758bffdff4f4ff5befd67d8aec278ca3cd6..0360f0d1ea8254633919bf56e6d83e050b1ac2c9 100644 --- a/modules/bert/finetuning-legal-bert-classifier.ipynb +++ b/modules/bert/finetuning-legal-bert-classifier.ipynb @@ -320,6 +320,8 @@ " precision = divide(tp, tp + fp)\n", " recall = divide(tp, tp + fn)\n", " f1 = divide(2 * precision * recall, precision + recall)\n", + " metrics[f\"recall_{id2label[label_idx]}\"] = recall\n", + " metrics[f\"precision_{id2label[label_idx]}\"] = precision\n", " metrics[f\"f1_{id2label[label_idx]}\"] = f1\n", "\n", " # (5)\n",