1 files changed, 11 insertions, 5 deletions
diff --git a/frontend/src/app/_data/Model.ts b/frontend/src/app/_data/Model.ts
index 216e1c36..a891c10c 100644
--- a/frontend/src/app/_data/Model.ts
+++ b/frontend/src/app/_data/Model.ts
@@ -1,13 +1,16 @@
 export default class Model {
+    _id: string = '';
     constructor(
         public name: string = 'Novi model',
         public description: string = '',
         public dateCreated: Date = new Date(),
-        public datasetId?: number,
+        public lastUpdated: Date = new Date(),
+        public datasetId: string = '',
 
-        //Test set settings
-        public inputColumns: number[] = [0],
-        public columnToPredict: number = 1,
+        // Test set settings
+        public inputColumns: string[] = [],
+        public columnToPredict: string = '',
+        public randomOrder: boolean = true,
         public randomTestSet: boolean = true,
         public randomTestSetDistribution: number = 0.10, //0.1-0.9 (10% - 90%)
 
@@ -22,7 +25,8 @@ export default class Model {
         public batchSize: number = 5,
         public inputLayerActivationFunction: ActivationFunction = ActivationFunction.Sigmoid,
         public hiddenLayerActivationFunction: ActivationFunction = ActivationFunction.Sigmoid,
-        public outputLayerActivationFunction: ActivationFunction = ActivationFunction.Sigmoid
+        public outputLayerActivationFunction: ActivationFunction = ActivationFunction.Sigmoid,
+        public username: string = ''
     ) { }
 }
 
@@ -31,6 +35,8 @@ export enum ANNType {
     Convolutional = 'konvoluciona'
 }
 
+// replaceMissing srednja vrednost mean, median, najcesca vrednost (mode)
+// removeOutliers
 export enum Encoding {
     Label = 'label',
     OneHot = 'one hot'