AutoML PipelineSuggester: don't recommend pipelines from first-stage …

…trainers that failed (#3593)
dotnet · Apr 26, 2019 · fe77854 · fe77854
1 parent 29d92ed
commit fe77854
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/Microsoft.ML.Auto/PipelineSuggesters/PipelineSuggester.cs b/src/Microsoft.ML.Auto/PipelineSuggesters/PipelineSuggester.cs
@@ -91,8 +91,8 @@ private static IEnumerable<SuggestedTrainer> GetTopTrainers(IEnumerable<Suggeste
             IEnumerable<SuggestedTrainer> 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<SuggestedPipelineRunDetail> sortedHistory = history.OrderBy(r => r.Score);