python - 如何将 pandas.core.series.Series 类型转换为二维数组？

Question

我正在尝试使用 KNNClassifier 训练模型。我将数据拆分如下：

X_train, X_test, y_train, y_test = train_test_split(X_bow, y, test_size=0.30, random_state=42)

y_train= y_train.astype('int')

neigh = KNeighborsClassifier(n_neighbors=3)
neigh.fit(X_train, y_train)

当我尝试测试它时，我得到一个值错误。

pre = neigh.predict(y_test)

Expected 2D array, got 1D array instead:
array=[0. 1. 1. ... 0. 0. 0.].
Reshape your data either using array.reshape(-1, 1) if your data has a 
single feature or array.reshape(1, -1) if it contains a single sample.

我的 y_test 是pandas.core.series.Series类型

那么如何将 pandas.core.series.Series 转换为 2D 数组以使此测试工作？

我试图将 y_test 转换为数据帧，然后转换为数组，但我得到另一个值错误并且我被卡住了。

y_test = pd.DataFrame(y_test)
y_test = y_test.as_matrix().reshape(-1,1)
pre = neigh.predict(y_test)

ValueError: Incompatible dimension for X and Y matrices: X.shape[1] == 1 while Y.shape[1] == 6038

标签： pythonmachine-learningscikit-learn

Answer 1

我猜你需要使用你的X_test变量/数组，而不是y_test.

X_test是independent用于测试我们模型准确性的变量/特征，是将与预测值进行比较的y_test实际值。target

例子：

pre = neigh.predict(X_test)

测量精度：

from sklearn.metrics import accuracy_score
accuracy_score(y_test, pre)