From c014dcc7f215a592376f523f7db1bdcfde3f7511 Mon Sep 17 00:00:00 2001 From: XiaoYun Zhang Date: Wed, 19 Jul 2023 12:23:10 -0700 Subject: [PATCH] fix issue --- .../AutoMLExperiment/Runner/SweepablePipelineRunner.cs | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/src/Microsoft.ML.AutoML/AutoMLExperiment/Runner/SweepablePipelineRunner.cs b/src/Microsoft.ML.AutoML/AutoMLExperiment/Runner/SweepablePipelineRunner.cs index 5cc5e5feb1..d32f931158 100644 --- a/src/Microsoft.ML.AutoML/AutoMLExperiment/Runner/SweepablePipelineRunner.cs +++ b/src/Microsoft.ML.AutoML/AutoMLExperiment/Runner/SweepablePipelineRunner.cs @@ -45,6 +45,10 @@ public TrialResult Run(TrialSettings settings) var models = new List(); foreach (var split in datasetSplit) { + // a work-around to fix issue https://github.com/dotnet/machinelearning-modelbuilder/issues/2718 + // where the root cause is the shape of deep learning model is determined by the first time when this model is trained + // therefore, the deep learning model can't be retrained using the same pipeline + mlnetPipeline = _pipeline.BuildFromOption(_mLContext, parameter); var model = mlnetPipeline.Fit(split.TrainSet); var eval = model.Transform(split.TestSet); metrics.Add(_metricManager.Evaluate(_mLContext, eval));