diff options
| author | TAMARA JERINIC <tamara.jerinic@gmail.com> | 2022-04-04 01:14:24 +0200 |
|---|---|---|
| committer | TAMARA JERINIC <tamara.jerinic@gmail.com> | 2022-04-04 01:14:24 +0200 |
| commit | 15a842c76f2d2200b9e5d1fd88f9cc755bac92cb (patch) | |
| tree | 63e617fc2618cfa43f22d3d282521bd06efb255f /backend/microservice | |
| parent | aa2a296ef27520f38957bf4ef103fd5571468937 (diff) | |
Izmena
Diffstat (limited to 'backend/microservice')
| -rw-r--r-- | backend/microservice/api/ml_service.py | 3 |
1 files changed, 3 insertions, 0 deletions
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') |