File tree Expand file tree Collapse file tree 3 files changed +6
-6
lines changed Expand file tree Collapse file tree 3 files changed +6
-6
lines changed Original file line number Diff line number Diff line change @@ -570,8 +570,8 @@ void CBoostedTreeFactory::initializeUnsetRegularizationHyperparameters(core::CDa
570
570
-mainLoopSearchInterval / 2.0 ,
571
571
mainLoopSearchInterval / 2.0 )
572
572
.value_or (fallback);
573
- m_SoftDepthLimitSearchInterval =
574
- max ( m_SoftDepthLimitSearchInterval, TVector{1.0 });
573
+ m_SoftDepthLimitSearchInterval = max (
574
+ m_SoftDepthLimitSearchInterval, TVector{MIN_SOFT_DEPTH_LIMIT });
575
575
LOG_TRACE (<< " soft depth limit search interval = ["
576
576
<< m_SoftDepthLimitSearchInterval.toDelimited () << " ]" );
577
577
m_TreeImpl->m_Regularization .softTreeDepthLimit (
Original file line number Diff line number Diff line change @@ -1417,7 +1417,7 @@ bool CBoostedTreeImpl::selectNextHyperparameters(const TMeanVarAccumulator& loss
1417
1417
scale * CTools::stableExp (parameters (i)));
1418
1418
break ;
1419
1419
case E_SoftTreeDepthLimit:
1420
- m_Regularization.softTreeDepthLimit (std::max ( parameters (i), 2.0 ));
1420
+ m_Regularization.softTreeDepthLimit (parameters (i));
1421
1421
break ;
1422
1422
case E_SoftTreeDepthTolerance:
1423
1423
m_Regularization.softTreeDepthTolerance (parameters (i));
Original file line number Diff line number Diff line change @@ -935,7 +935,7 @@ BOOST_AUTO_TEST_CASE(testCategoricalRegressors) {
935
935
LOG_DEBUG (<< " bias = " << modelBias);
936
936
LOG_DEBUG (<< " R^2 = " << modelRSquared);
937
937
BOOST_REQUIRE_CLOSE_ABSOLUTE (0.0 , modelBias, 0.16 );
938
- BOOST_TEST_REQUIRE (modelRSquared > 0.95 );
938
+ BOOST_TEST_REQUIRE (modelRSquared > 0.93 );
939
939
}
940
940
941
941
BOOST_AUTO_TEST_CASE (testFeatureBags) {
@@ -1354,7 +1354,7 @@ BOOST_AUTO_TEST_CASE(testBinomialLogisticRegression) {
1354
1354
LOG_DEBUG (<< " log relative error = "
1355
1355
<< maths::CBasicStatistics::mean (logRelativeError));
1356
1356
1357
- BOOST_TEST_REQUIRE (maths::CBasicStatistics::mean (logRelativeError) < 0.69 );
1357
+ BOOST_TEST_REQUIRE (maths::CBasicStatistics::mean (logRelativeError) < 0.70 );
1358
1358
meanLogRelativeError.add (maths::CBasicStatistics::mean (logRelativeError));
1359
1359
}
1360
1360
@@ -1587,7 +1587,7 @@ BOOST_AUTO_TEST_CASE(testMultinomialLogisticRegression) {
1587
1587
LOG_DEBUG (<< " log relative error = "
1588
1588
<< maths::CBasicStatistics::mean (logRelativeError));
1589
1589
1590
- BOOST_TEST_REQUIRE (maths::CBasicStatistics::mean (logRelativeError) < 2.1 );
1590
+ BOOST_TEST_REQUIRE (maths::CBasicStatistics::mean (logRelativeError) < 2.2 );
1591
1591
meanLogRelativeError.add (maths::CBasicStatistics::mean (logRelativeError));
1592
1592
}
1593
1593
You can’t perform that action at this time.
0 commit comments