@@ -963,13 +963,15 @@ BOOST_AUTO_TEST_CASE(testProgress) {
963
963
std::string phase{result[" phase_progress" ][" phase" ].GetString ()};
964
964
int progress{result[" phase_progress" ][" progress_percent" ].GetInt ()};
965
965
if (phase == maths::CBoostedTreeFactory::FEATURE_SELECTION) {
966
- featureSelectionLastProgress = progress;
966
+ featureSelectionLastProgress = std::max (featureSelectionLastProgress, progress) ;
967
967
} else if (phase == maths::CBoostedTreeFactory::COARSE_PARAMETER_SEARCH) {
968
- coarseParameterSearchLastProgress = progress;
968
+ coarseParameterSearchLastProgress =
969
+ std::max (coarseParameterSearchLastProgress, progress);
969
970
} else if (phase == maths::CBoostedTreeFactory::FINE_TUNING_PARAMETERS) {
970
- fineTuneParametersLastProgress = progress;
971
+ fineTuneParametersLastProgress =
972
+ std::max (fineTuneParametersLastProgress, progress);
971
973
} else if (phase == maths::CBoostedTreeFactory::FINAL_TRAINING) {
972
- finalTrainLastProgress = progress;
974
+ finalTrainLastProgress = std::max (finalTrainLastProgress, progress) ;
973
975
}
974
976
}
975
977
}
@@ -1052,17 +1054,19 @@ BOOST_AUTO_TEST_CASE(testProgressFromRestart) {
1052
1054
std::string phase{result[" phase_progress" ][" phase" ].GetString ()};
1053
1055
int progress{result[" phase_progress" ][" progress_percent" ].GetInt ()};
1054
1056
if (phase == maths::CBoostedTreeFactory::FEATURE_SELECTION) {
1055
- featureSelectionLastProgress = progress;
1057
+ featureSelectionLastProgress = std::max (featureSelectionLastProgress, progress) ;
1056
1058
} else if (phase == maths::CBoostedTreeFactory::COARSE_PARAMETER_SEARCH) {
1057
- coarseParameterSearchLastProgress = progress;
1059
+ coarseParameterSearchLastProgress =
1060
+ std::max (coarseParameterSearchLastProgress, progress);
1058
1061
} else if (phase == maths::CBoostedTreeFactory::FINE_TUNING_PARAMETERS) {
1059
1062
if (progress > 0 ) {
1060
1063
fineTuneParametersFirstProgress =
1061
1064
std::min (fineTuneParametersFirstProgress, progress);
1062
1065
}
1063
- fineTuneParametersLastProgress = progress;
1066
+ fineTuneParametersLastProgress =
1067
+ std::max (fineTuneParametersLastProgress, progress);
1064
1068
} else if (phase == maths::CBoostedTreeFactory::FINAL_TRAINING) {
1065
- finalTrainLastProgress = progress;
1069
+ finalTrainLastProgress = std::max (finalTrainLastProgress, progress) ;
1066
1070
}
1067
1071
}
1068
1072
}
0 commit comments