From 821ca89dd781f208da4a861a6d28017fe1dd89c2 Mon Sep 17 00:00:00 2001 From: Pavan Mandava Date: Mon, 27 Apr 2020 11:46:02 +0200 Subject: [PATCH] Added Result Model class --- eval/metrics.py | 13 +++++++++++-- 1 file changed, 11 insertions(+), 2 deletions(-) diff --git a/eval/metrics.py b/eval/metrics.py index 042296e..5ddcb15 100644 --- a/eval/metrics.py +++ b/eval/metrics.py @@ -1,6 +1,15 @@ +def f1_score(y_true, y_pred, labels, average): + return 0 -def f1_score(y_true, y_pred, labels, average): +class Result: - return 0 + def __init__(self, precision, recall, average, label): + self.precision = precision + self.recall = recall + self.average = average + self.label = label + self.f1_score = 2 * (precision * recall) / (precision + recall) + def print_result(self): + print('F1 Score :: ',self.f1_score)