Skip to content

Commit fe77854

Browse files
authored
AutoML PipelineSuggester: don't recommend pipelines from first-stage trainers that failed (#3593)
1 parent 29d92ed commit fe77854

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/Microsoft.ML.Auto/PipelineSuggesters/PipelineSuggester.cs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -91,8 +91,8 @@ private static IEnumerable<SuggestedTrainer> GetTopTrainers(IEnumerable<Suggeste
9191
IEnumerable<SuggestedTrainer> availableTrainers,
9292
bool isMaximizingMetric)
9393
{
94-
// narrow history to first stage runs
95-
history = history.Take(availableTrainers.Count());
94+
// narrow history to first stage runs that succeeded
95+
history = history.Take(availableTrainers.Count()).Where(x => x.RunSucceded);
9696

9797
history = history.GroupBy(r => r.Pipeline.Trainer.TrainerName).Select(g => g.First());
9898
IEnumerable<SuggestedPipelineRunDetail> sortedHistory = history.OrderBy(r => r.Score);

0 commit comments

Comments
 (0)