gpt4 book ai didi

python - 如何解决 ValueError : The number of classes has to be greater than one; got 1 class

转载 作者:行者123 更新时间:2023-12-05 06:01:07 43 4
gpt4 key购买 nike

当我运行以下命令时:

from sklearn.svm import SVC
from sklearn.model_selection import train_test_split
x_train, x_test, y_train, y_test = train_test_split(x, y, test_size=0.2)
clf = SVC(kernel='rbf', probability=True)
clf.fit(x_train, y_train)

我收到了 ValueError:类的数量必须大于一;上了 1 节课

我做了:print(np.unique(y_train)),它返回了 [0]

谁能指出我正确的解决方案方向?

最佳答案

要么您的 y 列表不包含 1,要么 y 中的 1 太少以至于 y_train 最终可能不包含 1。您应该打印 y,如果它包含 1,则需要更改拆分策略以确保所有类都出现在 y_trainy_test 至少一次

关于python - 如何解决 ValueError : The number of classes has to be greater than one; got 1 class,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/67307405/

43 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com