Python错误'int'对象没有属性'shape'

Question

使用下面的示例，sklearn 与 Python 3.5：

from sklearn import tree
features = [[140,1], [130,1], [150,0], [155,0]]
labels = [0,0,1,1]

clf = tree.DecisionTreeClassifier()
clf.fit(features, labels)
print(clf.predict(155,0))

我收到错误后的错误。我不明白为什么会收到此错误，有人可以解释一下吗？

Traceback (most recent call last):
  File "ml.py", line 7, in <module>
    print(clf.predict(155,0))
  File "/Users/user/Documents/hello/env/lib/python3.5/site-packages/sklearn/tree/tree.py", line 404, in predict
    X = self._validate_X_predict(X, check_input)
  File "/Users/user/Documents/hello/env/lib/python3.5/site-packages/sklearn/tree/tree.py", line 371, in _validate_X_predict
    n_features = X.shape[1]
AttributeError: 'int' object has no attribute 'shape'

Answer 1

如果您阅读 DecisionTreeClassifier.predict 的文档，您会发现您传递了错误的数据：

预测(X, check_input=True)

预测 X 的类或回归值。 对于分类模型，返回 X 中每个样本的预测类别。对于回归模型，返回基于 X 的预测值。 参数：

X : array-like or sparse matrix of shape = [n_samples, n_features]