在 DecisionTreeClassifier 中使用 class_weight 参数时有没有办法获取整数值

Question

我正在使用 DecisionTreeClassifier() 处理具有 class_weight='balanced' 参数的不平衡数据集。

绘制树时，节点在'value'属性中有浮动，我猜这与权重有关。我的问题是：这里有一种方法可以将这些值转换为实数（每个类的记录数）吗？或者至少他们可以总结出该节点上的样本数量？

我已经看到（在其他帖子中）在 RandomForest 中你可以设置 'bootstrap' 参数来改变它，我不知道是否与这里相同，但是 DecisionTree 类中没有引导参数。

Answer 1

如果您在 DecisionTreeClassifier() 中使用 class_weight='balanced'，这些值确实会出现加权（参见here）。

你可以做的是在 plot_tree 中使用 proportion=True，这至少会给你未加权的比例百分比。