windows 10 下安装机器学习利器 xgboost

xgboost在浅层机器学习中这么火,也来试试,发现网上安装xgboost的文章说的不少是模糊不清,笔者也是看了很多,其实很简单,这里来整理下,记录下windows 10 下安装xgboost的步骤

1、直接安装

pip install xgboost

有时由于网络原因,上面那条指令无法下载whl文件,笔者是没有成功安装

2、替代的方法安装

https://www.lfd.uci.edu/~gohlke/pythonlibs/#xgboost,在这个网站上找适合自己电脑的xgboost

3、GPU方案

从上面的两种方法安装后,并不是GPU版本,如果不想使用GPU版本,第三点儿可以忽略,需要的话继续往下看

在调用GPU时,有如下提示:

XGBoost version not compiled with GPU support

究其原因是编译的xgboost.dll 这个文件不是GPU版本的,可以从下面的链接中下载最新的GPU版本,覆盖这个dll即可

http://ssl.picnet.com.au/xgboost/

4、GPU版本使用方法

param['gpu_id'] = 0
param['max_bin'] = 16
param['tree_method'] = 'gpu_hist'
详细得使用可以参考下面的链接

https://xgboost.readthedocs.io/en/latest/gpu/index.html#
0
分享到:
广告位招租

评论1

请先

  1. 解决了我的问题你
    qqrew2019-08-07 10:57:08
没有账号? 忘记密码?