最近使用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分隔开的。