scikit learn train_test_split的使用

scikit learn train_test_split的使用

最近使用train_test_split，发现是一个高频的api，我分享一下我的代码，首先安装scikit-learn:

pip install scikit-learn

然后输入下面的代码：

import pandas as pdfrom sklearn.model_selection import train_test_splitdata_path='demo.csv'data=pd.read_csv(data_path,sep='\t')# print()# print(data.head())# questions=data["question"].tolist()train,test = train_test_split(data,test_size=0.2)

这段代码的意思是把demo.csv分为训练集和测试集两部分，demo.csv的数据使用\t分隔开的。

版权声明：本文内容由网络用户投稿，版权归原作者所有，本站不拥有其著作权，亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容，请联系我们jiasou666@gmail.com 处理，核实后本网站将在24小时内删除侵权内容。