From e9ba7063cd9daf3994338cdd4d79fe5ca6c23fb8 Mon Sep 17 00:00:00 2001
From: Julien Breton <xm9q8f80@jlnbrtn.me>
Date: Tue, 9 Jul 2024 18:07:32 +0200
Subject: [PATCH] add recall and precision on BERT
---
modules/bert/finetuning-legal-bert-classifier.ipynb | 2 ++
1 file changed, 2 insertions(+)
diff --git a/modules/bert/finetuning-legal-bert-classifier.ipynb b/modules/bert/finetuning-legal-bert-classifier.ipynb
index c104075..0360f0d 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",
--
GitLab