scikit learn train_test_split的使用

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

pip install scikit-learn   

然后输入下面的代码：

import pandas as pd
from sklearn.model_selection import train_test_split



data_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分隔开的。