From ec2eb6c1d39da383116affa73c3215944be638d5 Mon Sep 17 00:00:00 2001 From: Advaith S Rao Date: Tue, 2 Jan 2024 16:32:44 +0530 Subject: [PATCH] Update test_modeler.py Fixed unit test for macro f1 score in test_modeler --- tests/test_modeler.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_modeler.py b/tests/test_modeler.py index 9a175c1..9b1aed6 100644 --- a/tests/test_modeler.py +++ b/tests/test_modeler.py @@ -44,7 +44,7 @@ def mail(): def test_get_f1_score(y_true, y_pred): macro_f1_score = get_f1_score(y_true, y_pred, average='macro') weighted_f1_score = get_f1_score(y_true, y_pred, average='weighted') - assert round(macro_f1_score,3) == 0.667 + assert round(macro_f1_score,3) == 0.333 assert round(weighted_f1_score,3) == 0.667 def test_get_classification_report_confusion_matrix(y_true, y_pred):