需要一种简单的方法来列出数据集中的所有变量
发布于2022-12-01 07:34 阅读(189) 评论(0) 点赞(7) 收藏(2)
我正在尝试根据数据集在 python 中进行决策树分类。我的数据集有 149 个变量，标记为 V1、V2、...、V149。我宁愿不必为了制作树而输入所有这些变量。我不断收到一条错误消息，提示“功能名称的长度与功能的数量不匹配”。我在 Colabs 工作，但如果需要可以转到 Jupyter。
我试过了
from sklearn.tree import export_graphviz
from six import StringIO
from IPython.display import Image
import pydotplus
feature_cols = [**"Not sure what to say here"**]
dot_data = StringIO()
export_graphviz(clf, out_file=dot_data,
filled=True, rounded=True,
special_characters=True,feature_names = feature_cols,class_names=['0','1'])
graph = pydotplus.graph_from_dot_data(dot_data.getvalue())
graph.write_png('recidivism.png')
Image(graph.create_png())
解决方案
暂无回答
所属网站分类: 技术文章 > 问答
作者：黑洞官方问答小能手
链接：https://www.pythonheidong.com/blog/article/1870632/bb4714ce5dc74f8b24fe/
来源：python黑洞网
任何形式的转载都请注明出处,如有侵权 一经发现 必将追究其法律责任
昵称:
评论内容：(最多支持255个字符)
---无人问津也好，技不如人也罢，你都要试着安静下来，去做自己该做的事，而不是让内心的烦躁、焦虑，坏掉你本来就不多的热情和定力