【发布时间】: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