本文解决我问题时:xgboost在使用时原生接口和sklearn接口的异同
- 首先放上官方中文文档
2. xgboost的sklrean接口
from xgboost.sklearn import XGBClassifier
xgbc = XGBClassifier(n_jobs=-1) # 新建xgboost sklearn的分类class
# xgboost的sklearn接口默认只使用cpu单线程,设置n_jobs=-1使用所有线程
print("开始xgboost classifier训练")
xgbc.fit(train_vector,np.array(train_label))
# 喂给分类器训练numpy形式的训练特征向量和标签向量
print("完成xgboost classifier训练,开始预测")
pre_train_Classifier = xgbc.predict(test_vector) # 喂给分类器numpy形式的测试特征向量
np.save(os.path.join(model_path,"pre_train_Classifier.npy"),pre_train_Classifier) # 保存结果
xgboost的sklearn接口,可以不经过标签标准化(即将标签编