From 15a842c76f2d2200b9e5d1fd88f9cc755bac92cb Mon Sep 17 00:00:00 2001 From: TAMARA JERINIC Date: Mon, 4 Apr 2022 01:14:24 +0200 Subject: Izmena --- backend/microservice/api/ml_service.py | 3 +++ 1 file changed, 3 insertions(+) (limited to 'backend') diff --git a/backend/microservice/api/ml_service.py b/backend/microservice/api/ml_service.py index 40166cc4..73b191da 100644 --- a/backend/microservice/api/ml_service.py +++ b/backend/microservice/api/ml_service.py @@ -333,6 +333,9 @@ def train(dataset, params, callback): "adj_r2" : adj_r2 } elif(problem_type=="multi-klasifikacioni"): + + cr=sm.classification_report(y_test, y_pred) + cm=sm.confusion_matrix(y_test,y_pred) # https://www.kaggle.com/code/nkitgupta/evaluation-metrics-for-multi-class-classification/notebook accuracy=metrics.accuracy_score(y_test, y_pred) macro_averaged_precision=metrics.precision_score(y_test, y_pred, average = 'macro') -- cgit v1.2.3