有没有内置的方法让scikit学习执行随机分层k-fold交叉验证?这是最常见的CV方法之一,我很惊讶我找不到一个内置的方法来实现这一点。
我看到cross_validation.KFold()
有一个shuffling标志,但它不是分层的。不幸的是cross_validation.StratifiedKFold()
没有这样的选项,并且cross_validation.StratifiedShuffleSplit()
不会产生不相交的褶皱。
我遗漏了什么吗?这是计划好的吗?
(显然我可以自己实现)
Tags:
cross_validation.StratifiedKFold
的洗牌标志已在当前版本0.15中引入:http://scikit-learn.org/0.15/modules/generated/sklearn.cross_validation.StratifiedKFold.html
这可以在变更日志中找到:
http://scikit-learn.org/stable/whats_new.html#new-features
下面是我将分层洗牌分为训练集和测试集的实现:
此代码输出:
我想我会发布我的解决方案,以防对其他人有用。
相关问题 更多 >
编程相关推荐