【发布时间】:2019-12-15 14:14:50
【问题描述】:
我查看了其他答案,但仍然无法理解为什么问题仍然存在。
Iris 数据集的经典机器学习实践。
代码:
dataset=load_iris()
X = np.array(dataset.data)
y = np.array(dataset.target)
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.3)
model = KNeighborsClassifier()
model.fit(X_train, y_train)
prediction=model.predict(X_test)
所有数组的形状:
- X 形状:(150, 4)
- y 形状:(150,)
- X_train: (105, 4)
- X_test: (45, 4)
- y_train: (105,)
- y_test (45,)
- 预测:(45,)
尝试打印此 model.score(y_test, prediction),但我得到了错误。 我尝试使用 .reshape(-1,1) 将 y_test 和预测转换为 2D 数组,但出现另一个错误:查询数据维度必须与训练数据维度匹配。
这不仅关乎解决方案,还关乎了解问题所在。
【问题讨论】:
标签: python arrays numpy scikit-learn