train_test_split 分割测试集训练集

#导入模块包
from sklearn.model_selection import train_test_split
 
#分割数据, test_size 默认是0.75
x_train, x_test, y_train, y_test = train_test_split(x, y, test_size=0.7,random_state=1)
test_size 设置测试集的比例;
random_state 固定随机种子,便于多次测试时使用。

猜你喜欢

转载自www.cnblogs.com/chengziaichiyu/p/10352399.html