ValueError:n_splits = 10不能大于每个类中的成员数

ValueError:n_splits = 10不能大于每个类中的成员数,第1张

ValueError:n_splits = 10不能大于每个类中的成员

分层意味着在每个折叠中保持每个类的比率。因此,如果您的原始数据集有3个类别,比例分别为60%,20%和20%,那么分层将尝试在每个折叠中保持该比例。

就你而言

X = ["hey", "join now", "hello", "join today", "join us now", "not today",     "join this trial", " hey hey", " no", "hola", "bye", "join today",      "no","join join"]y = ["n", "r", "n", "r", "r", "n", "n", "n", "n", "y", "n", "n", "n", "y"]

您总共有14个样本(成员)与分布:

class    number of members         percentage 'n'        9  64 'r'        3  22 'y'        2  14

因此,StratifiedKFold将尝试在每次折叠中保持该比例。现在,您已指定10折(n_splits)。因此,对于y类,要保持此比率,至少要折叠2/10
= 0.2个成员。但是我们不能给出少于1个成员(样本),所以这就是为什么它会在其中抛出错误。

如果不是

n_splits=10
,则设置了
n_splits=2
,那么它会起作用,因为’y’的成员数量将是2/2 =1。为了
n_splits =10
正确工作,每个类至少需要有10个样本。



欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/zaji/5021203.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-11-15
下一篇 2022-11-15

发表评论

登录后才能评论

评论列表(0条)

保存