【问题标题】:GridSearchCV ranking not returning desired valuesGridSearchCV 排名未返回所需值
【发布时间】:2021-06-27 06:23:51
【问题描述】:

我正在做一个 NearestNeighbor 推荐模型,它采用单词列表并推荐相似的单词,我想调整 n_neighbors 的值。这是我输入的代码。

from sklearn.model_selection import GridSearchCV

gs_clf = GridSearchCV(NearestNeighbors(algorithm = 'brute'), {
    'n_neighbors': [1,2,3,4,5,6,7,8,9,10]
}, scoring = 'f1', cv=5)

gs_clf = gs_clf.fit(transformed_courses_new , np.array(courses.code))

trasnformed_courses_new 是一个形状为 (159, 120) 的数组,np.array(courses.code) 是 (159,) 并且每个值都是一个唯一的标签。所以我的理解是,gridsearch 将对 n_neighbors 的所有值进行测试,并对 f1 得分最高的 k 的最佳值进行排名。但是当我运行代码时,我收到了一个警告,即 NearestNeighbors 没有 .predict 功能。

有什么解决方法吗?

感谢任何帮助。

【问题讨论】:

    标签: machine-learning nearest-neighbor gridsearchcv


    【解决方案1】:

    使用KNeighborsClassifier

    (请记住,kNN 存在维度灾难。)

    【讨论】:

      猜你喜欢
      • 2016-11-22
      • 2013-03-19
      • 1970-01-01
      • 1970-01-01
      • 2021-09-27
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      相关资源
      最近更新 更多