出于可解释性的目的，我正在尝试在 python 中可视化决策树。我注意到像“NumGoals >= 1.23”这样的条件对用户来说可能非常模糊，我更愿意看到这样的东西：“NumGoals > 1”。

所以，我试图关闭归一化，我什至怀疑这可能是输入矩阵的dtype.

这是重现结果的简单代码。

from sklearn.datasets import load_iris
from sklearn.tree import DecisionTreeClassifier, plot_tree
import matplotlib.pyplot as plt
import numpy as np

iris = load_iris()

clf = DecisionTreeClassifier()
clf = clf.fit(np.array(iris.data, dtype=int), iris.target)
plot_tree(clf)
plt.show()

那么，甚至可以告诉树将整数视为整数吗？