KNN网格搜索优化

Question

我试图使用以下语法优化KNN模型

params = [{
    'knn_n_neighbors': [3, 5, 7, 9, 12,15,17],
    'knn_weights': ['uniform', 'distance'],
    \
    'knn_metric': ['euclidean', 'manhattan']}]

gs=GridSearchCV(KNN_model,param_grid=params, verbose=1, n_jobs=-1, cv=5)

当我做这个的时候-

gs_result=gs.fit(X_knn_train, y_knn_train)

我明白错误-

ValueError: Invalid parameter knn_metric for estimator KNeighborsClassifier(). Check the list of available parameters with `estimator.get_params().keys()`.

我做错了什么？有人能告诉我我不使用什么参数吗？

Answer 1

您需要使用以下键名创建params字典：

params = [{
    'n_neighbors': [3, 5, 7, 9, 12,15,17],
    'weights': ['uniform', 'distance'],
    \
    'metric': ['euclidean', 'manhattan']}]