ValueError：未知标签类型：'未知'

Question

我尝试运行以下代码。顺便说一句，我是python和sklearn的新手。

import pandas as pd
import numpy as np
from sklearn.linear_model import LogisticRegression


# data import and preparation
trainData = pd.read_csv('train.csv')
train = trainData.values
testData = pd.read_csv('test.csv')
test = testData.values
X = np.c_[train[:, 0], train[:, 2], train[:, 6:7],  train[:, 9]]
X = np.nan_to_num(X)
y = train[:, 1]
Xtest = np.c_[test[:, 0:1], test[:, 5:6],  test[:, 8]]
Xtest = np.nan_to_num(Xtest)


# model
lr = LogisticRegression()
lr.fit(X, y)

其中y是0和1的np.ndarray

我收到以下信息：

文件“ C：\ Anaconda3 \ lib \ site-packages \ sklearn \ linear_model \
logistic.py”，行> 1174，适合check_classification_targets（y）

文件“ C：\ Anaconda3 \ lib \ site-packages \ sklearn \ utils \
multiclass.py”，第172行，在check_classification_targets中引发ValueError（“未知标签类型：％r”％y_type）

ValueError：未知标签类型：’未知’

来自sklearn文档：http
://scikit-
learn.org/stable/modules/generated/sklearn.linear_model.LogisticRegression.html#sklearn.linear_model.LogisticRegression.fit

y：类数组，形状（n_samples），目标值（分类中的类标签，回归中的实数）

我怎么了

更新：

y是array（[0.0，1.0，1.0，…，0.0，1.0，0.0]，dtype = object）的大小是（891，）

Answer 1

您y的类型是object，因此sklearn无法识别其类型。在该行y=y.astype('int')之后添加该行y = train[:, 1]。