1 files changed, 5 insertions, 1 deletions
diff --git a/frontend/src/app/_elements/form-model/form-model.component.ts b/frontend/src/app/_elements/form-model/form-model.component.ts
index 82fa0e95..a98ceaec 100644
--- a/frontend/src/app/_elements/form-model/form-model.component.ts
+++ b/frontend/src/app/_elements/form-model/form-model.component.ts
@@ -149,5 +149,9 @@ export class FormModelComponent implements AfterViewInit {
     this.lossFunction = LossFunctionMultiClassification;
     this.newModel.lossFunction=LossFunction.SparseCategoricalCrossEntropy;
     }
-  }
+  
+}
+getInputColumns() {
+    return this.forExperiment.inputColumns.filter(x => x != this.forExperiment.outputColumn);
+}
 }