1 files changed, 7 insertions, 3 deletions
diff --git a/frontend/src/app/_elements/model-load/model-load.component.ts b/frontend/src/app/_elements/model-load/model-load.component.ts
index 6db40916..ca6b8ea5 100644
--- a/frontend/src/app/_elements/model-load/model-load.component.ts
+++ b/frontend/src/app/_elements/model-load/model-load.component.ts
@@ -1,9 +1,8 @@
-import { Component, OnInit } from '@angular/core';
+import { Component, OnInit, ViewChild, Output, EventEmitter } from '@angular/core';
 import Shared from 'src/app/Shared';
 import Model, { ActivationFunction, Encoding, LossFunction, LossFunctionBinaryClassification, LossFunctionMultiClassification, LossFunctionRegression, Metrics, MetricsBinaryClassification, MetricsMultiClassification, MetricsRegression, NullValueOptions, Optimizer, ProblemType } from 'src/app/_data/Model';
 import { ModelsService } from 'src/app/_services/models.service';
-import { Output } from '@angular/core';
-import { EventEmitter } from '@angular/core';
+import { GraphComponent } from '../graph/graph.component';
 
 
 @Component({
@@ -13,6 +12,7 @@ import { EventEmitter } from '@angular/core';
 })
 export class ModelLoadComponent implements OnInit {
 
+  @ViewChild(GraphComponent) graph!: GraphComponent;
   @Output() selectedModelChangeEvent = new EventEmitter<Model>();
 
   newModel: Model = new Model();
@@ -46,6 +46,10 @@ export class ModelLoadComponent implements OnInit {
   ngOnInit(): void {
   }
 
+  updateGraph() {
+    this.graph.update();
+  }
+
   getMetrics() {
     this.newModel.metrics = [];
     let cb = document.getElementsByName("cbmetrics");