在python中构建分类树时出错

时间:2014-10-23 09:22:44

标签: python machine-learning decision-tree

这是我在scikit learn website

中找到的代码
from sklearn.datasets import load_iris
from sklearn import tree
iris = load_iris()
clf = tree.DecisionTreeClassifier()
clf = clf.fit(iris.data, iris.target)

这里是什么(iris.data,iris.target), 为了我的项目目的使用这个代码,我修改了像    pest=open("project.csv","r+")

clf=clf.fit(pest.data,pest.target)

但是我得到一个错误文件没有属性“data”。 please click on this to view my data set "project"

我需要获得一个具有根节点作物和叶节点毒性水平的树。如何设置根节点和类标签。我已经尝试了很多,但无法得到它。

2 个答案:

答案 0 :(得分:0)

您的有害生物似乎只包含您的文件内容。您应该添加一个步骤,您可以选择文件的哪个部分是数据,哪个是目标。

答案 1 :(得分:0)

您似乎只打开.csv文件,并且不会从中加载任何内容。您需要加载数据并指定哪个列是'数据'哪个栏目是“目标”。 Here是一些用于从csv文件加载数据的示例代码。