在学knn最邻近分类时,原始数据根据参数Q1、Q2分为type=T1、type=T2两种;
执行
knn = neighbors.KNeighborsClassifier()
knn.fit(df[['Q1','Q2']],df['type'])
knn.predict([35,26])
ValueError: Expected 2D array, got 1D array instead:
array=[35 26].
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.
np.array([35,26]).reshape(1,-1)
结果为
之后,了解到把一维数组改为二维数组只要外层加一个[]就可以了,可以省去reshape。
knn = neighbors.KNeighborsClassifier()
knn.fit(df[['Q1','Q2']],df['type'])
knn.predict([[35,26]])