sklearn train_test_split用法说明

阅读量：

train_test_split用法详解

格式

复制代码

    def train_test_split(*arrays,
                     test_size=None,
                     train_size=None,
                     random_state=None,
                     shuffle=True,
                     stratify=None)

参数详解与分析

在这里插入图片描述

对stratify参数的进一步说明如下：

该参数的主要功能是确保在数据划分之后，训练集与测试集中各类别样本所占的比例与原始数据集保持一致。

例如，假设总共有100个样本，其中正类样本为10个，负类样本为90个，类别分布比例为1：10。若直接使用train_test_split()进行划分，训练集与验证

全部评论 (0)

还没有任何评论哟~