File tree Expand file tree Collapse file tree 1 file changed +3
-9
lines changed
test/Microsoft.ML.AutoML.Tests Expand file tree Collapse file tree 1 file changed +3
-9
lines changed Original file line number Diff line number Diff line change @@ -442,11 +442,9 @@ public void AutoFitWithPresplittedData()
442442
443443 }
444444
445- [ Theory , IterationData ( 10 ) ]
446- [ TestCategory ( "RunSpecificTest" ) ]
447- public void AutoFitMaxExperimentTimeTest ( int iteration )
445+ [ LightGBMFact ]
446+ public void AutoFitMaxExperimentTimeTest ( )
448447 {
449- Console . WriteLine ( iteration ) ;
450448 // A single binary classification experiment takes less than 5 seconds.
451449 // System.OperationCanceledException is thrown when ongoing experiment
452450 // is canceled and at least one model has been generated.
@@ -457,12 +455,8 @@ public void AutoFitMaxExperimentTimeTest(int iteration)
457455 var columnInference = context . Auto ( ) . InferColumns ( dataPath , DatasetUtil . UciAdultLabel ) ;
458456 var textLoader = context . Data . CreateTextLoader ( columnInference . TextLoaderOptions ) ;
459457 var trainData = textLoader . Load ( dataPath ) ;
460-
461- var experSettings = new BinaryExperimentSettings { MaxExperimentTimeInSeconds = 60 } ;
462- experSettings . Trainers . Remove ( BinaryClassificationTrainer . LightGbm ) ;
463-
464458 var experiment = context . Auto ( )
465- . CreateBinaryClassificationExperiment ( experSettings )
459+ . CreateBinaryClassificationExperiment ( 15 )
466460 . Execute ( trainData , new ColumnInformation ( ) { LabelColumnName = DatasetUtil . UciAdultLabel } ) ;
467461
468462 // Ensure the (last) model that was training when maximum experiment time was reached has been stopped,
You can’t perform that action at this time.
0 commit comments