zz962

  博客园 :: 首页 :: 博问 :: 闪存 :: 新随笔 :: 联系 :: 订阅 订阅 :: 管理 ::

引入

from sklearn.model_selection import train_test_split

分割(原数据是一个打他frame,先进行切片操作,分开数据和标签)

data_X = data.iloc[:,1:]
data_Y = data.iloc[:,0]

rate_test = 0.2  # 测试集比例
x_train, x_test, y_train, y_test = train_test_split(data_X,
                                                    data_Y,
                                                    test_size= rate_test)

  

posted on 2021-03-03 15:57  zz962  阅读(309)  评论(0编辑  收藏  举报