决策树graphviz可视化报错can only concatenate str (not “numpy.float64”) to str

最新推荐文章于 2025-07-16 01:00:00 发布

原创

最新推荐文章于 2025-07-16 01:00:00 发布 · 4.3k 阅读

9 ·

CC 4.0 BY-SA版权

文章标签：

#可视化 #决策树 #python #机器学习

在尝试使用graphviz对iris数据集的决策树进行可视化时，遇到'can only concatenate str (not "numpy.float64") to str'的错误。通过检查发现是将numpy浮点数转换为字符串的问题。修复代码后，成功生成了可视化的决策树，但初次显示异常。经过进一步调整，最终得到正常显示的决策树。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

决策树graphviz可视化报错can only concatenate str (not “numpy.float64”) to str

对iris数据集作的一个简单的决策树实例，但是在用graphviz可视化时报了“can only concatenate str (not “numpy.float64”) to str”的错误，之后成功解决，因为之前也出过一次bug，所以在此做个小小的总结。

修正前的代码：

from sklearn import tree
import graphviz
dot_data = tree.export_graphviz(model4, out_file = None,
                                feature_names = x_train.columns,
                                class_names = np.unique(y_train))
graph = graphviz.Source(dot_data)
graph.render("DecisionTree")      #在同级目录下生成DecisionTree.pdf文件

注：model4为已经生成的tree.DecisionTreeClassifier模型，x_train为训练集，y_train为训练集对应的结果标签，为0.，1.和2.，代表的是iris数据集中样本对应的类别。

因为输出np.unique(y_train)的时候得到的是array([0., 1., 2.])，所以将代码修改为：

dot_data = tree.export_graphviz(model4, out_file = None,
                                feature_names = x_train.columns,

最低0.47元/天解锁文章

200万优质内容无限畅学