aboutsummaryrefslogtreecommitdiff
path: root/frontend/src/app/_data
diff options
context:
space:
mode:
authorDanijel Andjelkovic <adanijel99@gmail.com>2022-03-31 14:09:42 +0200
committerDanijel Andjelkovic <adanijel99@gmail.com>2022-03-31 14:09:42 +0200
commit0515cd40b21caad00d375583d17c111bcc9d4635 (patch)
treefd3ad3bae117ef503f0dfbb70f30a274615aee16 /frontend/src/app/_data
parent8779b55a3a3637dbfb1bea518ebdc4d340687097 (diff)
parent355d4b6f2b2bfe6037533253d8d3e8428d59d992 (diff)
Merge branch 'dev' of http://gitlab.pmf.kg.ac.rs/igrannonica/neuronstellar into dev
# Conflicts: # frontend/src/app/_pages/add-model/add-model.component.html # frontend/src/app/_pages/add-model/add-model.component.ts
Diffstat (limited to 'frontend/src/app/_data')
-rw-r--r--frontend/src/app/_data/Model.ts32
1 files changed, 29 insertions, 3 deletions
diff --git a/frontend/src/app/_data/Model.ts b/frontend/src/app/_data/Model.ts
index fb6849e3..58bc3d4f 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
) { }
}
@@ -164,9 +164,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',
}