学习中心
登录
已解决
决策分类树可视化与GBDT分类可视化的问题
tree.plot_tree(model, filled=True, feature_names=feature_names, rounded=True, class_names=['0', '1', '2'])

996f6e54503758f41cdba85957a309d5.png

这是用决策分类树画的,这个图里面有指定数据进入哪一个子节点的类别

但是对于GBDT

tree.plot_tree(model1[0, 1], filled=True, feature_names=feature_names, rounded=True, class_names=['0', '1', '2'])

a697491dcc438f8ad42778700c0cfe03.png

却不会出现对应类别是什么原因?

是不是因为GBDT本质上用的是决策回归树,预测出来的是对应样本的估计值y_pred,再使用该y_pred带入sigmoid或者softmax函数计算对应概率值

得到概率值后才有根据在哪一个类别中概率大归为哪一类,所以画出来的图出来的是预测值而不是类别

54 2
    2个回答
    写回答