1 files changed, 6 insertions, 1 deletions
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);