Skip to content

Commit 3914d01

Browse files
committed
small changes for PR
1 parent 45078b7 commit 3914d01

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

test/Microsoft.ML.Benchmarks/RffTransform.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,7 @@ public void CV_Multiclass_Digits_RffTransform_OVAAveragedPerceptron()
4747
.AppendCacheCheckpoint(mlContext)
4848
.Append(mlContext.Transforms.Concatenate("Features", "FeaturesRFF"))
4949
.Append(new ValueToKeyMappingEstimator(mlContext, "Label"))
50-
.Append(mlContext.MulticlassClassification.Trainers.OneVersusAll(mlContext.BinaryClassification.Trainers.AveragedPerceptron(numberOfIterations: 10)));
50+
.Append(mlContext.MulticlassClassification.Trainers.OneVersusAll(mlContext.BinaryClassification.Trainers.AveragedPerceptron(numberOfIterations: 10), useProbabilities: false));
5151

5252
var cvResults = mlContext.MulticlassClassification.CrossValidate(data, pipeline, numberOfFolds: 5);
5353
}

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

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -220,9 +220,9 @@ public void MulticlassCVTest()
220220
[TestCategory("Multiclass")]
221221
public void MulticlassReductionTest()
222222
{
223-
RunOneAllTests(TestLearners.Ova, TestDatasets.iris, digitsOfPrecision: 6);
224-
RunOneAllTests(TestLearners.OvaWithFastForest, TestDatasets.iris, digitsOfPrecision: 6);
225-
RunOneAllTests(TestLearners.Pkpd, TestDatasets.iris, digitsOfPrecision: 6);
223+
RunOneAllTests(TestLearners.Ova, TestDatasets.iris, digitsOfPrecision: 5);
224+
RunOneAllTests(TestLearners.OvaWithFastForest, TestDatasets.iris, digitsOfPrecision: 5);
225+
RunOneAllTests(TestLearners.Pkpd, TestDatasets.iris, digitsOfPrecision: 5);
226226

227227
Done();
228228
}

0 commit comments

Comments
 (0)