Advertisement

安装xgboost到python环境

阅读量:

以下将对XGBoost的Python模块进行说明。

编译与导入Python模块

数据接口参数配置

模型训练过程

程序提前终止机制

预测操作流程

安装步骤

首先需安装XGBoost的C++版本,随后进入源文件根目录下的wrapper文件夹,并执行如下脚本以完成Python模块的安装:

python setup.py install

安装结束后,可采用以下方式导入XGBoost的Python模块:

import xgboost as xgb

数据接口功能说明

XGBoost支持加载libsvm格式的文本数据,所加载的数据形式可以是NumPy的二维数组或XGBoost自身的二进制缓存文件。这些数据将被存储在DMatrix对象中。

在加载libsvm格式的数据以及二进制缓存文件时,可使用如下方式:

drain = xgb.DMatrix('train.svm.txt')
dtest = xgb.DMatrix('test.svm.buffer')

当需要将NumPy数组载入DMatrix对象时,可使用以下方法:

data = np.random.rand(5,10) # 包含5个样本,每个样本有10个特征
label = np.random.randint(2, size=5) # 二分类目标变量
dtrain = xgb.DMatr

全部评论 (0)

还没有任何评论哟~