python划分数据集方法

可以使用 train_test_split() 函数将数据集划分为训练集和测试集。该函数使用随机采样的方式将数据集分配给不同的集合。示例代码如下：

from sklearn.model_selection import train_test_split
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)

其中，X 和 y 分别代表输入特征和标签，test_size 表示测试集大小占总数据集的比例，random_state 确定每次划分数据集时的随机种子，以保证结果的可重复性。