Skip to content

Commit 2c7f4b9

Browse files
Update AutoMLExperimentTests.cs
1 parent a35f7e2 commit 2c7f4b9

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

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

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -239,14 +239,12 @@ public async Task AutoMLExperiment_UCI_Adult_CV_5_Test()
239239
var experiment = context.Auto().CreateExperiment();
240240
var pipeline = context.Auto().Featurizer(data, "_Features_", excludeColumns: new[] { DatasetUtil.UciAdultLabel })
241241
.Append(context.Auto().BinaryClassification(DatasetUtil.UciAdultLabel, "_Features_", useLgbm: false, useSdcaLogisticRegression: false, useLbfgsLogisticRegression: false));
242-
var cts = new CancellationTokenSource();
243-
cts.CancelAfter(50000);
244242
experiment.SetDataset(data, 5)
245243
.SetBinaryClassificationMetric(BinaryClassificationMetric.AreaUnderRocCurve, DatasetUtil.UciAdultLabel)
246244
.SetPipeline(pipeline)
247-
.SetTrainingTimeInSeconds(100);
245+
.SetTrainingTimeInSeconds(10);
248246

249-
var result = await experiment.RunAsync(cts.Token);
247+
var result = await experiment.RunAsync();
250248
result.Metric.Should().BeGreaterThan(0.8);
251249
}
252250

0 commit comments

Comments
 (0)