diff --git a/src/Microsoft.ML.Auto/PipelineSuggesters/PipelineSuggester.cs b/src/Microsoft.ML.Auto/PipelineSuggesters/PipelineSuggester.cs index 38e79b9877..bf876bd775 100644 --- a/src/Microsoft.ML.Auto/PipelineSuggesters/PipelineSuggester.cs +++ b/src/Microsoft.ML.Auto/PipelineSuggesters/PipelineSuggester.cs @@ -91,8 +91,8 @@ private static IEnumerable GetTopTrainers(IEnumerable availableTrainers, bool isMaximizingMetric) { - // narrow history to first stage runs - history = history.Take(availableTrainers.Count()); + // narrow history to first stage runs that succeeded + history = history.Take(availableTrainers.Count()).Where(x => x.RunSucceded); history = history.GroupBy(r => r.Pipeline.Trainer.TrainerName).Select(g => g.First()); IEnumerable sortedHistory = history.OrderBy(r => r.Score);