From 64e6ba664ba7dc9da6b11ee951c1adadc817f722 Mon Sep 17 00:00:00 2001
From: Daniel Holstein <daholste@microsoft.com>
Date: Thu, 25 Apr 2019 15:17:46 -0700
Subject: [PATCH] AutoML PipelineSuggester: don't recommend pipelines from
 first-stage trainers that failed

---
 src/Microsoft.ML.Auto/PipelineSuggesters/PipelineSuggester.cs | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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<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);