Skip to content

Commit d42a522

Browse files
Changes for reviewer comments.
1 parent 483a50b commit d42a522

File tree

1 file changed

+17
-17
lines changed

1 file changed

+17
-17
lines changed

test/Microsoft.ML.Predictor.Tests/TestAutoInference.cs

Lines changed: 17 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -291,25 +291,25 @@ public void TestRocketPipelineEngine()
291291
[Fact(Skip = "Need CoreTLC specific baseline update")]
292292
public void TestTextDatasetLearn()
293293
{
294-
//using (var env = new TlcEnvironment())
295-
//{
296-
//string pathData = GetDataPath(@"../UnitTest/tweets_labeled_10k_test_validation.tsv");
297-
//int batchSize = 5;
298-
//int numIterations = 35;
299-
//int numTransformLevels = 1;
300-
//int numSampleRows = 100;
301-
//AutoInference.SupportedMetric metric = AutoInference.SupportedMetric.AccuracyMicro;
294+
using (var env = new TlcEnvironment())
295+
{
296+
string pathData = GetDataPath(@"../UnitTest/tweets_labeled_10k_test_validation.tsv");
297+
int batchSize = 5;
298+
int numIterations = 35;
299+
int numTransformLevels = 1;
300+
int numSampleRows = 100;
301+
AutoInference.SupportedMetric metric = AutoInference.SupportedMetric.AccuracyMicro;
302302

303-
//// Using the simple, uniform random sampling (with replacement) engine
304-
//PipelineOptimizerBase autoMlEngine = new UniformRandomEngine(env);
303+
// Using the simple, uniform random sampling (with replacement) engine
304+
PipelineOptimizerBase autoMlEngine = new UniformRandomEngine(env);
305305

306-
//// Test initial learning
307-
//var amls = AutoInference.InferPipelines(env, autoMlEngine, pathData, "", out var _, numTransformLevels, batchSize,
308-
//metric, out var _, numSampleRows, new IterationTerminator(numIterations),
309-
//MacroUtils.TrainerKinds.SignatureMultiClassClassifierTrainer);
310-
//env.Check(amls.GetAllEvaluatedPipelines().Length == numIterations);
311-
//}
312-
//Done();
306+
// Test initial learning
307+
var amls = AutoInference.InferPipelines(env, autoMlEngine, pathData, "", out var _, numTransformLevels, batchSize,
308+
metric, out var _, numSampleRows, new IterationTerminator(numIterations),
309+
MacroUtils.TrainerKinds.SignatureMultiClassClassifierTrainer);
310+
env.Check(amls.GetAllEvaluatedPipelines().Length == numIterations);
311+
}
312+
Done();
313313
}
314314

315315
[Fact(Skip = "Need CoreTLC specific baseline update")]

0 commit comments

Comments
 (0)