使用Azure ML SDK构建管道时,AutoML交叉验证是否自动分层?
Azure AutoML交叉验证的分层处理说明
- Azure AutoML针对分类任务执行交叉验证时,默认会自动对数据做分层处理,能保证每个交叉验证折叠(fold)里的标签分布和整个数据集的分布一致,刚好适配你这种标签0占30%、标签1占70%的不均衡场景。
- 这个分层逻辑是分类任务的默认行为,不需要额外配置,AutoML会自动基于目标标签列的类别完成分层拆分。但如果是回归任务,因为目标是连续值,不存在类别分层的逻辑,AutoML不会做分层处理。
如果需要自定义分层策略,或者想完全掌控数据拆分逻辑,你可以在传入AutoML前自行完成分层操作:
- 用
sklearn.model_selection.StratifiedKFold生成带分层逻辑的训练/验证折叠索引 - 在AutoML配置里把
cv参数设为你生成的自定义折叠索引,而不是直接指定折叠数量(比如cv=5)
内容的提问来源于stack exchange,提问作者Mavil
相关产品推荐
相关产品推荐

