python划分数据集

在 Python 中，可以使用 scikit-learn 库中的 train_test_split() 函数来划分数据集。该函数可以将数据集随机划分为训练集和测试集，也可以按照一定比例划分。以下是一个简单的示例代码：

from sklearn.model_selection import train_test_split

# 假设 data 是数据集，labels 是数据标签
X_train, X_test, y_train, y_test = train_test_split(data, labels, test_size=0.2, random_state=42)

其中，test_size 表示测试集所占比例，random_state 表示随机种子，用于保证每次划分结果一致。