标签: python python-3.x machine-learning scikit-learn
我在python3中使用XGBoost,我发现有两种方法可以“预测”predict()和prob_predict()。
predict()
prob_predict()
一等奖:获得概率最高的5个班级
银奖:获得前5个标签(概率)
铜奖:如何公开标签(Labelencoder?)并将其映射到prob_predict输出的概率
最简单的方法是修改sklearn.py并添加predict_n_ext(prediction_count=5)等函数
predict_n_ext(prediction_count=5)