Add XGBoost Classifier
This commit is contained in:
@@ -1,16 +1,11 @@
|
||||
#include "XGBoost.h"
|
||||
|
||||
|
||||
|
||||
|
||||
//See https ://stackoverflow.com/questions/36071672/using-xgboost-in-c
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
namespace pywrap {
|
||||
XGBoost::XGBoost() : PyClassifier("xgboost", "XGBClassifier")
|
||||
{
|
||||
validHyperparameters = { "tree_method", "early_stopping_rounds", "n_jobs" };
|
||||
}
|
||||
std::string XGBoost::version()
|
||||
{
|
||||
return callMethodString("1.0");
|
||||
|
@@ -5,7 +5,7 @@
|
||||
namespace pywrap {
|
||||
class XGBoost : public PyClassifier {
|
||||
public:
|
||||
XGBoost() : PyClassifier("xgboost", "XGBClassifier") {};
|
||||
XGBoost();
|
||||
~XGBoost() = default;
|
||||
std::string version();
|
||||
};
|
||||
|
Reference in New Issue
Block a user