XGBoost获取带有标签的类

时间:2016-11-18 11:00:33

标签: python python-3.x machine-learning scikit-learn

我在python3中使用XGBoost,我发现有两种方法可以“预测”predict()prob_predict()

一等奖:获得概率最高的5个班级

银奖:获得前5个标签(概率)

铜奖:如何公开标签(Labelencoder?)并将其映射到prob_predict输出的概率

最简单的方法是修改sklearn.py并添加predict_n_ext(prediction_count=5)等函数

0 个答案:

没有答案