程序不工作“TypeError：fit（）缺少1个必需的位置参数：'y'”

Question

from sklearn import tree
from sklearn.datasets import load_iris
iris=load_iris()
dir(iris)
#output data to traixn setosa,versicolor and virginica
x=iris.data
#fetching data
x=np.delete(x, np.s_[::50], 0)
#print(x)
y=iris.target
#featching output
y=np.delete(y, np.s_[::50], 0)
algo=tree.DecisionTreeClassifier

当我尝试使用fit时它不支持

train=algo.fit(x,y)
res=train.pridict([test_setosa])
print(res)

Answer 1

您需要更改代码中的内容。 DecisionTreeClassifier是一个类，您在代码中调用它的方式是错误的。

<强>替换

algo=tree.DecisionTreeClassifier

<强>与

algo=tree.DecisionTreeClassifier()

完整代码

from sklearn import tree
from sklearn.datasets import load_iris
import numpy as np

iris=load_iris()
dir(iris)
#output data to traixn setosa,versicolor and virginica
x=iris.data
#fetching data
x=np.delete(x, np.s_[::50], 0)
#print(x)
y=iris.target
#featching output
y=np.delete(y, np.s_[::50], 0)

algo=tree.DecisionTreeClassifier()

train=algo.fit(x,y)
res=train.predict([test_setosa])