Merge branch 'dev' of http://gitlab.pmf.kg.ac.rs/igrannonica/neuronstellar into dev

1 files changed, 8 insertions, 2 deletions

diff --git a/frontend/src/app/_data/Model.ts b/frontend/src/app/_data/Model.ts
index 32247bbd..7a82cc5c 100644
--- a/frontend/src/app/_data/Model.ts
+++ b/frontend/src/app/_data/Model.ts
@@ -28,7 +28,7 @@ export default class Model {
         public outputLayerActivationFunction: ActivationFunction = ActivationFunction.Sigmoid,
         public username: string = '',
         public nullValues: NullValueOptions = NullValueOptions.DeleteRows,
-        public nullValuesReplacers = [],
+        public nullValuesReplacers: NullValReplacer[] = [],
         public metrics: Metric[] = [], // TODO add to add-model form
         public epochs: number = 5 // TODO add to add-model form
     ) { }
@@ -115,9 +115,15 @@ export enum ReplaceWith {
     Median = 'Medijana'
 }
 
+export class NullValReplacer {
+    "column": string;
+    "option": NullValueOptions;
+    "value" : any;
+}
+
 export enum Metric {
     MSE = 'mse',
     MAE = 'mae',
     RMSE = 'rmse'
     //...
-}
\ No newline at end of file
+}