【发布时间】:2021-08-11 13:37:37
【问题描述】:
我有一个 K 最近邻分类器,您可以在下面看到。据我了解,GridSearchCV 正在使用 1-20 之间的不同 k 值测试模型。当我做y_pred=knn_grid_cv.predict(x_test) 时,我得到了一堆 y 预测,但是使用什么值 k(1-20 之间)来获得这些 y 预测?它会是 GridSearchCV 中得分最高的 k 值吗?
x=football_df["Pace"].values.reshape(-1, 1)
print(x)
y=football_df["Position"].values.reshape(-1, 1)
x_train, x_test, y_train, y_test = train_test_split(x,y,test_size=0.4,random_state=42)
param_grid={"n_neighbors":np.arange(1,20)}
knn = KNeighborsClassifier()
knn_grid_cv = GridSearchCV(knn, param_grid, cv=5)
knn_grid_cv.fit(x_train,y_train)
y_pred=knn_grid_cv.predict(x_test)
print(y_pred)
【问题讨论】:
标签: python machine-learning scikit-learn grid-search hyperparameters