ValueError:类的数量必须大于一（python）

Question

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

回溯（最近一次呼叫最后一次）：

文件“C:/Classify/classifier.py”，第95行，在

列车平均值，测试平均值，cms=列车模型（X，y，“ceps”，plot=True）
列车模型中第47行的文件“C:/Classify/classifier.py”

clf.fit（X_train，y_train）文件“C:\Python27\lib\site packages\sklearn\svm\base.py”，第676行，在fit raise ValueError（“类数必须大于”ValueError：类数必须大于一。

下面是我的代码：

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

确切地最后一列（标签）只有一种类型（分类）。你至少应该有两个。例如如果标签决定是否必须卸载，则标签列应具有卸载和不卸载或（0或1）。

Answer 2

在当前的培训集中，您可能只有一个唯一的类标签。正如错误消息所指出的，数据集中至少需要有两个唯一的类。例如，您可以运行np.unique（y）查看数据集中的唯一类标签是什么。