XGBoost 入门使用方法

来源:互联网 发布:java api文档英文版载 编辑:程序博客网 时间:2024/04/30 18:50
  • 官方网址:http://xgboost.readthedocs.io/en/latest/
  • 安装步骤(ubuntu):
    • 下载C++源码: git clone –recursive https://github.com/dmlc/xgboost
    • 多进程编译 安装: cd xgboost; make -j4
    • 安装 python 支持: cd python-packages; sudo python setup.py install
  • 测试是否安装成功:

    import xgboost as xgb# read in datadtrain = xgb.DMatrix('demo/data/agaricus.txt.train')dtest = xgb.DMatrix('demo/data/agaricus.txt.test')# specify parameters via mapparam = {'max_depth':2, 'eta':1, 'silent':1, 'objective':'binary:logistic' }num_round = 2bst = xgb.train(param, dtrain, num_round)# make predictionpreds = bst.predict(dtest)

    执行上述代码,demo在xgboost目录下,如果没有问题,则安装成功。

原创粉丝点击