Skip to content

Commit 483a50b

Browse files
More changes for reviewer commenrts.
1 parent 76dd270 commit 483a50b

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

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

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -73,7 +73,7 @@ public void EntryPointPipelineSweepSerialization()
7373
var pathData = GetDataPath(@"../../Samples/UCI/adult.train");
7474
var pathDataTest = GetDataPath(@"../../Samples/UCI/adult.test");
7575
const int numOfSampleRows = 1000;
76-
//int numIterations = 10;
76+
int numIterations = 10;
7777
const string schema =
7878
"sep=, col=Features:R4:0,2,4,10-12 col=workclass:TX:1 col=education:TX:3 col=marital_status:TX:5 col=occupation:TX:6 " +
7979
"col=relationship:TX:7 col=ethnicity:TX:8 col=sex:TX:9 col=native_country:TX:13 col=label_IsOver50K_:R4:14 header=+";
@@ -131,8 +131,9 @@ public void EntryPointPipelineSweepSerialization()
131131

132132
var results = runner.GetOutput<IDataView>("ResultsOut");
133133
Assert.NotNull(results);
134-
//var rows = PipelinePattern.ExtractResults(Env, results, "Graph", "MetricValue", "PipelineId");
135-
//Assert.True(rows.Length == numIterations);
134+
var rows = PipelinePattern.ExtractResults(Env, results,
135+
"Graph", "MetricValue", "PipelineId", "TrainingMetricValue", "FirstInput", "PredictorModel");
136+
Assert.True(rows.Length == numIterations);
136137
}
137138

138139
[Fact]

0 commit comments

Comments
 (0)