python - 如何强制决策树在评估时仅使用整数

Question

我正在做一个决策树，我想强制算法仅以整数来评估条件。我使用的功能是离散和整数，因此树显示 "X <= 42.5" 没有意义，因此在本例中，我希望树显示在框中输入 “X < 43” 或 “X <= 42” 中的等效项之一。

我需要这个来让非技术人员更容易理解这棵树。显示“少于 15.5 首歌曲”没有意义，应该少于 43 或 小于或等于42.

我尝试更改源表的列类型，都是int64，问题依旧。

我正在使用的代码:

clf = tree.DecisionTreeClassifier(criterion='gini',
                                  max_depth=2,
                                  splitter='best')

clf = clf.fit(users_data, users_target)

到目前为止，我在文档中没有找到任何参数或类似的内容。

谢谢!

Answer 1

首先，我不会自己调整树规则，我会调整情节。

还有一个额外的tree ploting package来自 sklearn 。通过此调整:

precision : int, optional (default=3)

    Number of digits of precision for floating point in the values of impurity, threshold and value attributes of each node.

您可以更改它，例如:

tree.plot_tree(clf, precision=0)

应该给你四舍五入的数字。