3 files changed, 38 insertions, 5 deletions
diff --git a/frontend/src/app/_data/Dataset.ts b/frontend/src/app/_data/Dataset.ts
new file mode 100644
index 00000000..6e6cbffe
--- /dev/null
+++ b/frontend/src/app/_data/Dataset.ts
@@ -0,0 +1,15 @@
+export default class Dataset {
+    _id: string = '';
+    constructor(
+        public name: string = 'Novi izvor podataka',
+        public description: string = '',
+        public header: string[] = [],
+        public fileId?: number,
+        public extension: string = '.csv',
+        public isPublic: boolean = false,
+        public accessibleByLink: boolean = false,
+        public dateCreated: Date = new Date(),
+        public lastUpdated: Date = new Date(),
+        public username: string = 'tester1'
+    ) { }
+}
+\ No newline at end of file
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'
diff --git a/frontend/src/app/_data/Predictor.ts b/frontend/src/app/_data/Predictor.ts
new file mode 100644
index 00000000..7e902eae
--- /dev/null
+++ b/frontend/src/app/_data/Predictor.ts
@@ -0,0 +1,12 @@
+export default class Predictor {
+    _id: string = '';
+    constructor(
+        public name: string = 'Novi prediktor',
+        public description: string = '',
+        public inputs: string[] = [],
+        public output: string = '',
+        public isPublic: boolean = false,
+        public accessibleByLink: boolean = false,
+        public dateCreated: Date = new Date()
+    ) { }
+}
+\ No newline at end of file