train_test_split获取一定数量的数据
时间: 2023-10-19 19:26:25 浏览: 22
train_test_split可以用来从给定的数据集中获取一定数量的数据。它通过设置参数test_size来指定测试集的大小,即所需的数据数量。例如,如果将test_size设置为0.2,那么将从给定的数据集中获取20%的数据作为测试集,剩下的80%的数据将用于训练。此外,还可以使用参数stratify来保持split前类的分布。这样可以确保训练集和测试集中的类别比例与原始数据集中的类别比例相同。<span class="em">1</span><span class="em">2</span><span class="em">3</span>
#### 引用[.reference_title]
- *1* *2* *3* [2021大三机器学习:train_test_split方法详解](https://blog.csdn.net/Xmumu_/article/details/120677767)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 100%"]
[ .reference_list ]