Skip to content

Commit a18b9cb

Browse files
Update AutoMLExperimentTests.cs to fix timeout error (#6638)
1 parent 3386245 commit a18b9cb

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

test/Microsoft.ML.AutoML.Tests/AutoMLExperimentTests.cs

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -223,7 +223,7 @@ public async Task AutoMLExperiment_UCI_Adult_Train_Test_Split_Test()
223223
experiment.SetDataset(context.Data.TrainTestSplit(data))
224224
.SetBinaryClassificationMetric(BinaryClassificationMetric.AreaUnderRocCurve, DatasetUtil.UciAdultLabel)
225225
.SetPipeline(pipeline)
226-
.SetTrainingTimeInSeconds(1);
226+
.SetMaxModelToExplore(1);
227227

228228
var result = await experiment.RunAsync();
229229
result.Metric.Should().BeGreaterThan(0.8);
@@ -248,7 +248,7 @@ public async Task AutoMLExperiment_UCI_Adult_CV_5_Test()
248248
experiment.SetDataset(data, 5)
249249
.SetBinaryClassificationMetric(BinaryClassificationMetric.AreaUnderRocCurve, DatasetUtil.UciAdultLabel)
250250
.SetPipeline(pipeline)
251-
.SetTrainingTimeInSeconds(10);
251+
.SetMaxModelToExplore(1);
252252

253253
var result = await experiment.RunAsync();
254254
result.Metric.Should().BeGreaterThan(0.8);
@@ -275,7 +275,7 @@ public async Task AutoMLExperiment_Iris_CV_5_Test()
275275
experiment.SetDataset(data, 5)
276276
.SetMulticlassClassificationMetric(MulticlassClassificationMetric.MacroAccuracy, label)
277277
.SetPipeline(pipeline)
278-
.SetTrainingTimeInSeconds(10);
278+
.SetMaxModelToExplore(1);
279279

280280
var result = await experiment.RunAsync();
281281
result.Metric.Should().BeGreaterThan(0.8);
@@ -302,7 +302,7 @@ public async Task AutoMLExperiment_Iris_Train_Test_Split_Test()
302302
experiment.SetDataset(context.Data.TrainTestSplit(data))
303303
.SetMulticlassClassificationMetric(MulticlassClassificationMetric.MacroAccuracy, label)
304304
.SetPipeline(pipeline)
305-
.SetTrainingTimeInSeconds(10);
305+
.SetMaxModelToExplore(1);
306306

307307
var result = await experiment.RunAsync();
308308
result.Metric.Should().BeGreaterThan(0.8);

0 commit comments

Comments
 (0)