安装
Mac El Captain下安装xgboost:
git clone --recursive https://github.com/dmlc/xgboost
- cd xgboost; cp make/minimum.mk ./config.mk; make -j4
- cd python-package/
- 注释掉setup.py中的include_package_data=True
- sudo python setup.py install # 这样以后在ipynb中就能用了
- cd xgboost; touch VERSION
Starter Code
import xgboost as xgb
# read in data
dtrain = xgb.DMatrix('demo/data/agaricus.txt.train')
dtest = xgb.DMatrix('demo/data/agaricus.txt.test')
# specify parameters via map
param = {'max_depth':2, 'eta':1, 'silent':1, 'objective':'binary:logistic' }
num_round = 2
bst = xgb.train(param, dtrain, num_round)
# make prediction
preds = bst.predict(dtest)
DMatrix(some_pd_DataFrame, label=)
重要的param
“objective”:”count:poisson”, #分类/回归/排序等
“eval_metric”:[ “mae”]