機器學習實戰(zhàn)九天老師菜菜老師訓練營
2、數(shù)據(jù)集劃分(70 % ~80%訓練 + 20%~30%測試)
sklearn.model_selection.train_test_split(arrays, *options)
(1) 數(shù)據(jù)集的特征值
(2) 數(shù)據(jù)集的標簽值
(3) test_size測試集的大小,通常是比例e.g 0.2
(4) random_state 隨機數(shù)種子:如果是不同算法對比,要使得種子一致 -> 劃分一致
(5) return 訓練集特征值x_train,測試集特征值x_test,訓練集目標值y_train,測試集目標值y_test
標簽: