1 files changed, 34 insertions, 8 deletions
diff --git a/frontend/src/app/training/training.component.ts b/frontend/src/app/training/training.component.ts
index cb6c304c..4f20bc87 100644
--- a/frontend/src/app/training/training.component.ts
+++ b/frontend/src/app/training/training.component.ts
@@ -1,29 +1,55 @@
 import { Component, OnInit } from '@angular/core';
+import Shared from '../Shared';
 import Experiment from '../_data/Experiment';
 import Model from '../_data/Model';
+import { DatasetsService } from '../_services/datasets.service';
+import { ExperimentsService } from '../_services/experiments.service';
+import { ModelsService } from '../_services/models.service';
 
 @Component({
   selector: 'app-training',
   templateUrl: './training.component.html',
   styleUrls: ['./training.component.css']
 })
-export class TrainingComponent implements OnInit {
-
-  constructor() { }
-
-  ngOnInit(): void {
-  }
+export class TrainingComponent{
 
+  myExperiments?: Experiment[];
   selectedExperiment?: Experiment;
   selectedModel?: Model;
 
   trainingResult: any;
 
-  selectModel($model: Model) {
+  term: string = "";
+
+  constructor(private modelsService: ModelsService, private datasetsService: DatasetsService, private experimentsService: ExperimentsService) { 
+    this.experimentsService.getMyExperiments().subscribe((experiments) => {
+      this.myExperiments = experiments;
+      console.log(this.myExperiments);
+    });
+  }
+
+  selectThisExperiment(experiment: Experiment) {
+    this.selectedExperiment = experiment;
+  }
 
+  selectModel(model: Model) {
+    this.selectedModel = model;
   }
 
   trainModel() {
-    //eksperiment i model moraju da budu izabrani
+    this.trainingResult = undefined;
+
+    if (this.selectedExperiment == undefined) {
+      Shared.openDialog("Greška", "Molimo Vas da izaberete eksperiment iz kolekcije.");
+      return;
+    }
+    if (this.selectedModel == undefined) {
+      Shared.openDialog("Greška", "Molimo Vas da izaberete model.");
+      return;
+    }
+    this.modelsService.trainModel(this.selectedModel._id, this.selectedExperiment._id).subscribe((response: any) => {
+      console.log('Train model complete!', response);
+      this.trainingResult = response;
+    });
   }
 }