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

Question

我正在使用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[]这样的特征名称： 我该如何使实际的特征名称显示出来？（例如，Cape而不是X[3]等）

- IvanHid

Answer 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:])

- Anne