1 files changed, 20 insertions, 5 deletions

diff --git a/frontend/src/app/_data/Predictor.ts b/frontend/src/app/_data/Predictor.ts
index 8aa2b6cb..55d610ed 100644
--- a/frontend/src/app/_data/Predictor.ts
+++ b/frontend/src/app/_data/Predictor.ts
@@ -1,13 +1,28 @@
-export default class Predictor {
-    _id: string = '';
+import { FolderFile } from "./FolderFile";
+
+export default class Predictor extends FolderFile {
     constructor(
-        public name: string = 'Novi prediktor',
+        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(),
-        public uploaderId: string = ''
+        dateCreated: Date = new Date(),
+        lastUpdated: Date = new Date(),
+        public uploaderId: string = '',
+        //public finalMetrics: Metric[] = []
+        public experimentId: string = "",
+        public modelId: string = "",
+    ) {
+        super(name, dateCreated, lastUpdated);
+    }
+}
+
+export class Metric {
+    constructor(
+        public name: string = '',
+        public jsonValue: string = ''
     ) { }
+
 }
\ No newline at end of file