From 355d4b6f2b2bfe6037533253d8d3e8428d59d992 Mon Sep 17 00:00:00 2001
From: TAMARA JERINIC <tamara.jerinic@gmail.com>
Date: Thu, 31 Mar 2022 13:55:39 +0200
Subject: Dodat je automatski izbor metrika.

---
 frontend/src/app/_data/Model.ts | 32 +++++++++++++++++++++++++++++---
 1 file changed, 29 insertions(+), 3 deletions(-)

(limited to 'frontend/src/app/_data')

diff --git a/frontend/src/app/_data/Model.ts b/frontend/src/app/_data/Model.ts
index ff9f8329..1a16d960 100644
--- a/frontend/src/app/_data/Model.ts
+++ b/frontend/src/app/_data/Model.ts
@@ -31,7 +31,7 @@ export default class Model {
         public username: string = '',
         public nullValues: NullValueOptions = NullValueOptions.DeleteRows,
         public nullValuesReplacers: NullValReplacer[] = [],
-        public metrics: Metric[] = [], // TODO add to add-model form
+        public metrics: string[] = [], // TODO add to add-model form
         public epochs: number = 5 // TODO add to add-model form
     ) { }
 }
@@ -167,9 +167,35 @@ export class NullValReplacer {
     "value" : string;
 }
 
-export enum Metric {
+export enum Metrics {
     MSE = 'mse',
     MAE = 'mae',
     RMSE = 'rmse'
-    //...
+    
+}
+export enum MetricsRegression
+{
+    Mse = 'mse',
+    Mae = 'mae',
+    Mape = 'mape',
+    Msle='msle',
+    CosineProximity='cosine'
+}
+export enum MetricsBinaryClassification
+{
+    Accuracy='binary_accuracy',
+    Auc="AUC",
+    Precision='precision_score',
+    Recall='recall_score',
+    F1='f1_score',
+    
+
+}
+export enum MetricsMultiClassification
+{
+    Accuracy='categorical_accuracy',
+    Auc="AUC",
+    Precision='precision_score',
+    Recall='recall_score',
+    F1='f1_score',
 }
-- 
cgit v1.2.3