Make xgboost version same a sklearn
This commit is contained in:
@@ -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 */
|
@@ -7,7 +7,6 @@ namespace pywrap {
|
||||
public:
|
||||
XGBoost();
|
||||
~XGBoost() = default;
|
||||
std::string version();
|
||||
};
|
||||
} /* namespace pywrap */
|
||||
#endif /* XGBOOST_H */
|
Reference in New Issue
Block a user