XGBoost 入门使用方法

本文介绍了XGBoost在Ubuntu系统上的安装步骤及测试方法。包括下载C++源码、多进程编译安装、安装Python支持等内容,并通过一个示例验证了安装的成功。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

  • 官方网址: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 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)
    

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

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值