ValueError：未知标签类型：“未知”-标签是数字

Question

我正在为二进制分类问题构建一个随机森林分类器。我的标签都是数字的。

print labels.unique()
[1 0]

print type(labels)
    <class 'pandas.core.series.Series'>
print labels.shape
(3000,)

但是当我用 Gridsearchcv 拟合模型时

pipeline = Pipeline(steps=[('scaler', scaler), ('algorithm', algo)])
cv = StratifiedShuffleSplit(labels, 5, test_size=0.25, random_state=42)
gs = GridSearchCV(pipeline, param_grid, cv=cv, scoring='f1')
gs.fit(features, labels)

我收到了这个错误

ValueError: Unknown label type: 'unknown'

但是当我使用

gs.fit(features, labels.astype(int))

一切正常。谁能告诉我标签的问题出在哪里？

Answer 1

您只需使用tolist() 方法更改要列出的标签类型。使用

labels_lst = labels.tolist()

Scikit-learn 无法自动将系列转换为标签列表。