1.随机抽样方法
手动分离数据集
以房价数据集为例 8:2的比例分配训练集和测试集
import numpy as np
def split_train_test(data, test_ratio):
# 数据集shape[0]长度的随机序列
np.random.seed(33) # 确保生成序列一致,保持每次生成数据集一直
indices = np.random.permutation(len(data))
test_size = int(len(data) * test_ratio)
test_indices = indices[:test_size] # 前百分之test_ratio为测试集
train_indices = indices[test_size:]
return data.iloc[train_indices]