1 files changed, 22 insertions, 0 deletions
diff --git a/frontend/src/app/_data/Experiment.ts b/frontend/src/app/_data/Experiment.ts
index 706231c7..10320ab6 100644
--- a/frontend/src/app/_data/Experiment.ts
+++ b/frontend/src/app/_data/Experiment.ts
@@ -6,7 +6,29 @@ export default class Experiment {
         public datasetId: string = '',
         public inputColumns: string[] = [],
         public columnToPredict: string = '',
+        public nullValues: NullValueOptions = NullValueOptions.DeleteRows,
+        public nullValuesReplacers: NullValReplacer[] = [],
         public dateCreated: Date = new Date(),
         public lastUpdated: Date = new Date()
     ) { }
+}
+
+export enum NullValueOptions {
+    DeleteRows = 'delete_rows',
+    DeleteColumns = 'delete_columns',
+    Replace = 'replace'
+}
+
+export enum ReplaceWith {
+    None = 'Popuni...',
+    Mean = 'Srednja vrednost',
+    Median = 'Medijana',
+    Min = 'Minimum',
+    Max = 'Maksimum'
+}
+
+export class NullValReplacer {
+    "column": string;
+    "option": NullValueOptions;
+    "value": string;
 }
 \ No newline at end of file