2 files changed, 7 insertions, 5 deletions
diff --git a/frontend/src/app/_data/Experiment.ts b/frontend/src/app/_data/Experiment.ts
index 23fd77d4..ab87e3ca 100644
--- a/frontend/src/app/_data/Experiment.ts
+++ b/frontend/src/app/_data/Experiment.ts
@@ -13,10 +13,7 @@ export default class Experiment {
         public lastUpdated: Date = new Date(),
         public modelIds: string[] = [],
 
-        // Test set settings
-        public randomOrder: boolean = true,
-        public randomTestSet: boolean = true,
-        public randomTestSetDistribution: number = 0.1, //0.1-0.9 (10% - 90%) JESTE OVDE ZAKUCANO 10, AL POSLATO JE KAO 0.1 BACK-U
+        
 
         public encodings: ColumnEncoding[] = []//[{columnName: "", columnEncoding: Encoding.Label}]
     ) { }
diff --git a/frontend/src/app/_data/Model.ts b/frontend/src/app/_data/Model.ts
index 6281748c..185e2257 100644
--- a/frontend/src/app/_data/Model.ts
+++ b/frontend/src/app/_data/Model.ts
@@ -23,7 +23,12 @@ export default class Model extends FolderFile {
         public epochs: number = 5, // TODO add to add-model form
         public inputColNum: number = 5,
         public learningRate: LearningRate = LearningRate.LR1,
-        public layers: Layer[] = [new Layer()]
+        public layers: Layer[] = [new Layer()],
+
+        // Test set settings
+        public randomOrder: boolean = true,
+        public randomTestSet: boolean = true,
+        public randomTestSetDistribution: number = 0.1 //0.1-0.9 (10% - 90%) JESTE OVDE ZAKUCANO 10, AL POSLATO JE KAO 0.1 BACK-U
 
     ) {
         super(name, dateCreated, lastUpdated);