如何在 Graphviz 中显示特征名称？答案

【问题标题】：How to show Feature Names in Graphviz?如何在 Graphviz 中显示特征名称？
【发布时间】：2018-07-07 16:19:25
【问题描述】：

我正在 Graphviz 中构建一棵树，但我似乎无法显示特征名称，我已经定义了一个包含特征名称的列表，如下所示：

names = list(df.columns.values)

哪些打印：

    ['Gender',
 'SuperStrength',
 'Mask',
 'Cape',
 'Tie',
 'Bald',
 'Pointy Ears',
 'Smokes']

所以列表正在创建，稍后我会像这样构建树：

 export_graphviz(tree, out_file=ddata, filled=True, rounded=True, special_characters=False, impurity=False, feature_names=names)

但最终的图像仍然有像 X[] 这样的特征名称：

如何才能显示实际的功能名称？（Cap而不是X[3]等）

【问题讨论】：

【解决方案1】：

我只能想象这与将名称作为值数组传递有关。如果您直接传递列，它可以正常工作：

export_graphviz(tree, out_file=ddata, filled=True, rounded=True, special_characters=False, impurity=False, feature_names=df.columns)

如果需要，您还可以对列进行切片：

export_graphviz(tree, out_file=ddata, filled=True, rounded=True, special_characters=False, impurity=False, feature_names=df.columns[5:])

【讨论】：