使用scikit learn的决策树分类器存在一些问题

Question

我正在尝试使用sci kit学习实现决策树分类器。我从sklearn导入了它，但.fit()显示了一些问题。

我正在编写代码：

from sklearn.datasets import load_iris from sklearn import tree

iris = load_iris()

clf = tree.DecisionTreeClassifier()

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

with open("iris.dot", 'w') as f:

   f = tree.export_graphviz(clf, out_file=f)
import os

os.unlink('iris.dot')

以下是错误消息：

  

文件“”，第5行，in clf = clf.fit（iris.data，iris.target）

     

文件“C：\ Users \ ChirantanGupta \ Downloads \ WinPython-64bit-3.5.3.0Qt5 \ python-3.5.3.amd64 \ lib \ site-packages \ sklearn \ tree \ tree.py”，第739行，适合X_idx_sorted = X_idx_sorted）

     

文件“C：\ Users \ ChirantanGupta \ Downloads \ WinPython-64bit-3.5.3.0Qt5 \ python-3.5.3.amd64 \ lib \ site-packages \ sklearn \ tree \ tree.py”，第173行，适合getattr（y，“dtype”，无）！= DOUBLE或不y.flags.contiguous：

     

NameError：未定义名称“DOUBLE”

我无法理解这个错误。怎么修复？任何人都可以建议。