调整特定树节点的树参数

Question

我使用来自sklearn的DecisionTreeClassifier

我需要更正特定树节点中使用的拆分器（功能）min_samples_leaf。

我该怎么做？

Answer 1

您不能为单个节点定义min_samples_leaf，因为模型最终分配给其他节点的样本可能会少于整个模型的min_samples_leaf，以确保符合适用于该单个节点的规则。

如果您正在处理不平衡的数据集，我建议您在输入模型之前oversample or undersample您的数据，或者您可以手动设置类权重。

根据scikit-learn's user guide：

在训练之前平衡您的数据集，以防止树被 偏向于占主导地位的阶级。类平衡可以 通过从每个类中抽取相同数量的样本来完成，或 最好通过标准化样本权重的总和 (sample_weight) 为每个类设置相同的值。