值错误: n_splits=3 不能大于每个类中的成员数



我收到一个错误:

ValueError: n_splits=3 cannot be greater than the number of members in each class.

在这一行中:

gs_clf_svm = gs_clf_svm.fit(X, y)
y.shape
Out[148]: (6,)
y
Out[149]: 
array(['Andheri East', 'Goregaon', 'Powai', 'Andheri East', 'Goregaon',
'Powai'], dtype=object)

上行是数据和形状详细信息的详细信息。请建议

对于目标,每个唯一类只有 2 个成员。尝试将n_splits设置为 2 表示GridSearchCV。或者增加数据集。

最新更新