From fed12b26ae71e7a95d2dd1f4703541138a780d75 Mon Sep 17 00:00:00 2001 From: Antonio Velazquez Date: Thu, 17 Dec 2020 10:46:43 -0800 Subject: [PATCH] RunSpecific in test --- test/Microsoft.ML.AutoML.Tests/AutoFitTests.cs | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/test/Microsoft.ML.AutoML.Tests/AutoFitTests.cs b/test/Microsoft.ML.AutoML.Tests/AutoFitTests.cs index d5af1f1be5..c85e939a32 100644 --- a/test/Microsoft.ML.AutoML.Tests/AutoFitTests.cs +++ b/test/Microsoft.ML.AutoML.Tests/AutoFitTests.cs @@ -442,9 +442,11 @@ public void AutoFitWithPresplittedData() } - [LightGBMFact] - public void AutoFitMaxExperimentTimeTest() + [Theory, IterationData(10)] + [TestCategory("RunSpecificTest")] + public void AutoFitMaxExperimentTimeTest(int iteration) { + Console.WriteLine(iteration); // A single binary classification experiment takes less than 5 seconds. // System.OperationCanceledException is thrown when ongoing experiment // is canceled and at least one model has been generated. @@ -455,8 +457,12 @@ public void AutoFitMaxExperimentTimeTest() var columnInference = context.Auto().InferColumns(dataPath, DatasetUtil.UciAdultLabel); var textLoader = context.Data.CreateTextLoader(columnInference.TextLoaderOptions); var trainData = textLoader.Load(dataPath); + + var experSettings = new BinaryExperimentSettings { MaxExperimentTimeInSeconds = 60}; + experSettings.Trainers.Remove(BinaryClassificationTrainer.LightGbm); + var experiment = context.Auto() - .CreateBinaryClassificationExperiment(15) + .CreateBinaryClassificationExperiment(experSettings) .Execute(trainData, new ColumnInformation() { LabelColumnName = DatasetUtil.UciAdultLabel }); // Ensure the (last) model that was training when maximum experiment time was reached has been stopped,