Make xgboost version same a sklearn

This commit is contained in:
2024-01-18 10:23:24 +01:00
parent 071f7167e6
commit 7e7a9299fc
2 changed files with 1 additions and 6 deletions

View File

@@ -2,12 +2,8 @@
//See https ://stackoverflow.com/questions/36071672/using-xgboost-in-c
namespace pywrap {
XGBoost::XGBoost() : PyClassifier("xgboost", "XGBClassifier")
XGBoost::XGBoost() : PyClassifier("xgboost", "XGBClassifier", true)
{
validHyperparameters = { "tree_method", "early_stopping_rounds", "n_jobs" };
}
std::string XGBoost::version()
{
return callMethodString("1.0");
}
} /* namespace pywrap */

View File

@@ -7,7 +7,6 @@ namespace pywrap {
public:
XGBoost();
~XGBoost() = default;
std::string version();
};
} /* namespace pywrap */
#endif /* XGBOOST_H */