python中train_test_split函数的功能有哪些
lewis
2016-12-26
21次阅读
train_test_split函数的功能有:
- 将数据集分割成训练集和测试集两部分,可以指定分割比例。
- 可以根据需要随机打乱数据集。
- 可以根据指定的随机种子确保每次运行时分割结果一致。
- 可以同时分割多个特征和目标变量。
- 可以根据需要对分割后的数据进行stratify分层抽样,以保持原数据集中类别比例的一致性。

发表评论:
◎欢迎参与讨论,请在这里发表您的看法、交流您的观点。