您的位置:首页 > 编程语言 > Python开发

python 把数据分成训练集和测试集

2018-10-30 11:07 1161 查看
[code]from sklearn.model_selection import train_test_split
import pandas as pd

f1=pd.read_excel('aaa.xlsx')

f1.columns
#Index(['X', 'Y'], dtype='object')

X_train,X_test,y_train,y_test = train_test_split(X,Y,test_size=0.3)

X_train: 训练集X
y_train: 训练集Y
X_test:测试集X
y_test:测试集Y
train_test_split:切割样本
test_size=0.3:选取样本30%作为测试集的数量

 

阅读更多
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: