Skip to content

Commit e46e70b

Browse files
authored
[ML] Test threshold fix (#1007)
1 parent 309db87 commit e46e70b

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

lib/api/unittest/CDataFrameAnalyzerTrainingTest.cc

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -752,7 +752,7 @@ BOOST_AUTO_TEST_CASE(testRunBoostedTreeClassifierImbalanced) {
752752
api::CDataFrameAnalyzer analyzer{
753753
test::CDataFrameAnalysisSpecificationFactory::predictionSpec(
754754
test::CDataFrameAnalysisSpecificationFactory::classification(),
755-
"target", numberExamples, 4, 17000000, 0, 0, {"target"}),
755+
"target", numberExamples, 4, 18000000, 0, 0, {"target"}),
756756
outputWriterFactory};
757757

758758
TStrVec actuals;
@@ -800,7 +800,7 @@ BOOST_AUTO_TEST_CASE(testCategoricalFields) {
800800
api::CDataFrameAnalyzer analyzer{
801801
test::CDataFrameAnalysisSpecificationFactory::predictionSpec(
802802
test::CDataFrameAnalysisSpecificationFactory::regression(),
803-
"x5", 1000, 5, 26000000, 0, 0, {"x1", "x2"}),
803+
"x5", 1000, 5, 27000000, 0, 0, {"x1", "x2"}),
804804
outputWriterFactory};
805805

806806
TStrVec x[]{{"x11", "x12", "x13", "x14", "x15"},
@@ -906,7 +906,7 @@ BOOST_AUTO_TEST_CASE(testCategoricalFieldsEmptyAsMissing) {
906906
api::CDataFrameAnalyzer analyzer{
907907
test::CDataFrameAnalysisSpecificationFactory::predictionSpec(
908908
test::CDataFrameAnalysisSpecificationFactory::classification(),
909-
"x5", 1000, 5, 26000000, 0, 0, {"x1", "x2", "x5"}),
909+
"x5", 1000, 5, 27000000, 0, 0, {"x1", "x2", "x5"}),
910910
outputWriterFactory};
911911

912912
TStrVec fieldNames{"x1", "x2", "x3", "x4", "x5", ".", "."};

0 commit comments

Comments
 (0)