python - 参数 (n_neighbors) 的参数值需要是一个序列

Question

我正在尝试使用 Skleanr 模块。但是，我的代码在下面

n_range = {'n_neighbors': range(1,100)}
knn_search = GridSearchCV(estimator = KNeighborsClassifier(), param_grid=n_range, scoring='f1_micro')
knn_search.fit(features_vector, train_labels)

导致错误:

Parameter values for parameter (n_neighbors) need to be a sequence.

我做错了什么？

Answer 1

在 Python 3.x 中，函数 range 返回一个 range 对象(不是序列)，而不是列表。您必须自己将其转换为列表:

n_range = {'n_neighbors': list(range(1,100))}