eXtreme Gradient Boosting (GBDT, GBRT or GBM) Library for large-scale and distributed machine learning, on single node, hadoop yarn and more.
xgboost
after a global install with sudo python setup.py install
? I didn't realize there was a pip installable package already and I'd rather do that in my virtualenv
xgb = XGBRegressor(n_estimators=1000, silent=0)
xgb.fit(train.as_matrix(), trainY, verbose=1, eval_metric="rmse")
P = xgb.predict(test.as_matrix())