1 files changed, 8 insertions, 1 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 1d38de68..663e414f 100644
--- a/frontend/src/app/_elements/model-load/model-load.component.ts
+++ b/frontend/src/app/_elements/model-load/model-load.component.ts
@@ -1,7 +1,8 @@
-import { Component, OnInit } from '@angular/core';
+import { Component, OnInit, ViewChild } 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 { GraphComponent } from '../graph/graph.component';
 
 @Component({
   selector: 'app-model-load',
@@ -10,6 +11,8 @@ import { ModelsService } from 'src/app/_services/models.service';
 })
 export class ModelLoadComponent implements OnInit {
 
+  @ViewChild(GraphComponent) graph!: GraphComponent;
+
   newModel: Model = new Model();
 
   ProblemType = ProblemType;
@@ -33,6 +36,10 @@ export class ModelLoadComponent implements OnInit {
   ngOnInit(): void {
   }
 
+  updateGraph() {
+    this.graph.update();
+  }
+
   getMetrics() {
     this.newModel.metrics = [];
     let cb = document.getElementsByName("cbmetrics");