ValueError:类的数量必须大于一个(Python)

Question

在fit中传递x,y时，我会收到以下错误：

追溯（最近的最新电话）：

文件" c：/classify/classifier.py"，第95行，

train_avg，test_avg，cms = train_model（x，y，" ceps"，plot = true）
文件" c：/classify/classifier.py"，第47行，in train_model

clf.fit（x_train，y_train）文件" c： python27 lib lib site-packages sklearn svm svm base.py"，第676行， 提高价值（"类必须大于" ValueRror：类的数量必须大于一个。

）。

以下是我的代码：

def train_model(X, Y, name, plot=False):
"""
    train_model(vector, vector, name[, plot=False])
    Trains and saves model to disk.
"""
labels = np.unique(Y)
cv = ShuffleSplit(n=len(X), n_iter=1, test_size=0.3, indices=True, random_state=0)
train_errors = []
test_errors = []
scores = []
pr_scores = defaultdict(list)
precisions, recalls, thresholds = defaultdict(list), defaultdict(list), defaultdict(list)
roc_scores = defaultdict(list)
tprs = defaultdict(list)
fprs = defaultdict(list)
clfs = []  # for the median
cms = []
for train, test in cv:
    X_train, y_train = X[train], Y[train]
    X_test, y_test = X[test], Y[test]
    clf = LogisticRegression()
    clf.fit(X_train, y_train)
    clfs.append(clf)

Answer 1

您可能只有一个唯一的类标签。正如错误消息所指出的那样，您需要在数据集中至少有两个唯一的类。例如，您可以运行np.unique(y)来查看数据集中的唯一类标签。

Answer 2

准确。您的最后一列（标签）只有一种类型（分类）。您应该至少有两个。例如;如果您的标签是决定要么必须卸载，则标签列应具有卸载和Notoverload或（0或1）。