为什么scikit-learn中没有定义`model.fit`?

时间:2017-01-06 19:14:00

标签: python scikit-learn

我正在关注this示例的第3步:

?

我不明白为什么scikit不承认model.fit(dataset.data, dataset.target) expected = dataset.target predicted = model.predict(dataset.data)

我是否需要先分配该变量? 是否缺少导入?

我在jupyter工作,scikit-learn 0.17.1。

2 个答案:

答案 0 :(得分:1)

您需要首先启动您正在使用的任何模型的实例:

from sklearn.tree import DecisionTreeClassifier
model = DecisionTreeClassifier()    
model.fit(dataset.data, dataset.target)

答案 1 :(得分:0)

fit(x,y)是一种可用于估算器的方法。

为了能够在model上使用此方法,您必须首先创建模型并确保其为估算器类。

Documentation