小工具知识网

当前位置: 首页 > 行业知识 > sklearn函数:train

行业知识

sklearn函数:train

2026-11-13 11:49:52 来源:互联网转载

���数的功能是分割训练集和测试集

()

函数中参数的用法:

1. test_size:float or int, default=None

测试集的大小,如果是小数的话,值在(0,1)之间,表示测试集所占有的比例;

()

如果是整数,表示的是测试集的具体样本数;

如果train_size也是None的话,会有一个默认值0.25

2. train_size:float or int, default=None

和test_size一样,同上

3.random_state:int or RandomState instance, default=None

这个参数表示随机状态,因为每次分割都是随机的,

设置为整数则每次分割不随机

注意:只有当shuffle=True时,random_state才起作用

4.shuffle: bool, default=True

是否重洗数据(洗牌),就是说在分割数据前,是否把数据打散重新排序这样子,看上面我们分割完的数据,都不是原始数据集的顺序,默认是要重洗的

5.stratify:array-like, default=None

这个要配合结果集使用,假设原始的结果集中有2种分类,A:B=1:2

我们在随机分配的时候,是无法保证训练集和测试集中的A与B的比例

这时候,就可以用这个参数了,设置stratify=y,就可以让测试集和训练集中的结果集也保证这种分布

上一篇:Hololens2远程音视频通话与AR远程空间标注,基于OpenXR+MRTK3+WebRTC实现

下一篇:神威服务器在哪里?