程序员最近都爱上了这个网站  程序员们快来瞅瞅吧!  it98k网:it98k.com

本站消息

站长简介/公众号

  出租广告位,需要合作请联系站长

+关注
已关注

分类  

暂无分类

标签  

暂无标签

日期归档  

暂无数据

需要一种简单的方法来列出数据集中的所有变量

发布于2022-12-01 07:34     阅读(155)     评论(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黑洞网

任何形式的转载都请注明出处,如有侵权 一经发现 必将追究其法律责任

7 0
收藏该文
已收藏

评论内容:(最多支持255个字符)