diff options
| author | TAMARA JERINIC <tamara.jerinic@gmail.com> | 2022-05-12 12:04:42 +0200 |
|---|---|---|
| committer | TAMARA JERINIC <tamara.jerinic@gmail.com> | 2022-05-12 12:04:42 +0200 |
| commit | 166bceb500bd3475c196b7aa9ad9f68c4f83cefc (patch) | |
| tree | 3f9397853f0062852fcdeaac2d19a108ac663352 /backend/microservice/api | |
| parent | e8908768803c0fe037f5e873003078cc7cdb6293 (diff) | |
Izmena poziva funkcije.
Diffstat (limited to 'backend/microservice/api')
| -rw-r--r-- | backend/microservice/api/newmlservice.py | 16 |
1 files changed, 7 insertions, 9 deletions
diff --git a/backend/microservice/api/newmlservice.py b/backend/microservice/api/newmlservice.py index c401a3e6..6a863013 100644 --- a/backend/microservice/api/newmlservice.py +++ b/backend/microservice/api/newmlservice.py @@ -384,17 +384,15 @@ def train(dataset, paramsModel,paramsExperiment,paramsDataset,callback): classifier.save(filepath, save_format='h5') - - accuracy=metrics.accuracy_score(y_test, y_pred) - macro_averaged_precision=metrics.precision_score(y_test, y_pred, average = 'macro') - micro_averaged_precision=metrics.precision_score(y_test, y_pred, average = 'micro') - macro_averaged_recall=metrics.recall_score(y_test, y_pred, average = 'macro') - micro_averaged_recall=metrics.recall_score(y_test, y_pred, average = 'micro') - macro_averaged_f1=metrics.f1_score(y_test, y_pred, average = 'macro') - micro_averaged_f1=metrics.f1_score(y_test, y_pred, average = 'micro') + metrics={} + macro_averaged_precision=sm.precision_score(y_test, y_pred, average = 'macro') + micro_averaged_precision=sm.precision_score(y_test, y_pred, average = 'micro') + macro_averaged_recall=sm.recall_score(y_test, y_pred, average = 'macro') + micro_averaged_recall=sm.recall_score(y_test, y_pred, average = 'micro') + macro_averaged_f1=sm.f1_score(y_test, y_pred, average = 'macro') + micro_averaged_f1=sm.f1_score(y_test, y_pred, average = 'micro') metrics= { - "accuracy" : float(accuracy), "macro_averaged_precision" :float(macro_averaged_precision), "micro_averaged_precision" : float(micro_averaged_precision), "macro_averaged_recall" : float(macro_averaged_recall), |