windows下python安装xgboost

时间:2022-08-29 03:32:08

简介

最近要用到xgboost,但是python的官网并没有直接提供windows版本。其github源码 听说为了方便版本控制,删除了之前的window分支,统一使用CMake进行版本控制,因此在window平台上装xgboost Python安装包就比以前稍微麻烦了点,下面将提供两种方式可以安装xgboost,第一种方式不需要本地编译,第二种方式需要下载源码进行本地编译。

依赖包

在安装xgboost前需要安装以下工具包
1. Numpy
2. Scipy

通过Wheel安装xgboost(不需要本地编译)

有人提供了python非官方的windows平台包,下载对应的wheel文件后,在当前目录打开cmd输入以下命令

pip install xxx.whl

然后就可以打开python解释器输入

import xgboost as xgb

验证是否成功安装xgboost包

本地编译源码

本地编译源码可以通过MinGW和Visual Studio 编译。但是在编译前需要安装以下软件:
1. Git For Windows
2. CMake
3. 本地编译器(MinGW or Visual Studio)

Git For Windows

git主要是用来下载源码,打开git,输入下面命令可以下载源码

$ git clone –recursive https://github.com/dmlc/xgboost

CMake

CMake主要用来生成Visual Studio的工程文件,因为源码是通过CMake管理的,CMake是一个跨平台的构建工程的软件。在下载的xgboost文件夹里面打开git输入一下命令:

$ mkdir build
$ cd build
$ cmake .. -G”Visual studio 14 2015 win64”

因为我电脑上用的VS 2015,你可以根据你电脑上VS的版本修改-G”Visual studio 14 2015 win64”这个选项。

本地编译

进入build文件夹后会看到.sln项目解决方案。用VS打开编译就可以了

安装xgboost 包

在下载的xgboost目下的python-package文件夹里打开cmd,输入下面的命令:

python setup.py install

执行上述命令,如果没用错误的话就安装成功了,这时候打开python解释器就可以导入xgboost包看看有没有安装成功