Skip to content

Commit 7b6af06

Browse files
fix issue (#6768)
1 parent 65c7ca9 commit 7b6af06

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

src/Microsoft.ML.AutoML/AutoMLExperiment/Runner/SweepablePipelineRunner.cs

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -45,6 +45,10 @@ public TrialResult Run(TrialSettings settings)
4545
var models = new List<ITransformer>();
4646
foreach (var split in datasetSplit)
4747
{
48+
// a work-around to fix issue https://github.com/dotnet/machinelearning-modelbuilder/issues/2718
49+
// where the root cause is the shape of deep learning model is determined by the first time when this model is trained
50+
// therefore, the deep learning model can't be retrained using the same pipeline
51+
mlnetPipeline = _pipeline.BuildFromOption(_mLContext, parameter);
4852
var model = mlnetPipeline.Fit(split.TrainSet);
4953
var eval = model.Transform(split.TestSet);
5054
metrics.Add(_metricManager.Evaluate(_mLContext, eval));

0 commit comments

Comments
 (0)