[sklearn] KFold与StratifiedKFold用法
1. 概览
KFold和StratifiedKFold的作用都是用于配合交叉验证的需求,将数据分割成训练集和测试集。
2. 区别
KFold随机分割数据,不会考虑数据的分布情况。
StratifiedKFold会根据原始数据的分布情况,分割出同分布的数据。
3. 实验
3.1 代码
from sklearn.model_selection import KFold
from sklearn.model_selection import StratifiedKFold
import numpy as np
原创
2020-12-21 16:40:21 ·
766 阅读 ·
0 评论