划分训练集和测试集python代码

以下是一个简单的 Python 代码示例，用于将数据集分成训练集和测试集：

import numpy as np

# 载入整个数据集
data = np.loadtxt('data.csv', delimiter=',')

# 随机打乱数据
np.random.shuffle(data)

# 划分训练集和测试集的比例
train_ratio = 0.8

# 计算训练集和测试集的数量
train_size = int(train_ratio * len(data))
test_size = len(data) - train_size

# 划分训练集和测试集
train_data = data[:train_size]
test_data = data[train_size:]

print(f'Train data size: {len(train_data)}')
print(f'Test data size: {len(test_data)}')

上述代码中，假设数据存储在名为 `data.csv` 的 CSV 文件中。首先，我们使用 `numpy` 库来将数据加载到一个名为 `data` 的 NumPy 数组中。然后，我们使用 `np.random.shuffle()` 函数来随机打乱数据顺序。接下来，我们使用 `train_ratio` 变量指定训练集所占比例，计算出训练集和测试集的数量。最后，我们使用 NumPy 数组切片来提取训练集和测试集，并打印它们的大小。

相关问题

划分训练集和测试集python代码实现

在Python中，我们可以使用sklearn库中的train_test_split函数来划分训练集和测试集。具体的实现代码如下所示：

from sklearn.model_selection import train_test_split

# X为特征数据，y为标签数据，test_size为测试集占比
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.3, random_state=42)

其中，X为特征数据，y为标签数据，test_size表示测试集占总数据集的比例，random_state为随机数种子，用于保证每次划分的结果一致。划分后，我们可以使用X_train和y_train来训练模型，使用X_test和y_test来测试模型性能。

划分训练集和测试集的python代码

以下是使用sklearn库中train_test_split方法划分训练集和测试集的Python代码：

from sklearn.model_selection import train_test_split

#加载数据集，X为数据特征，y为数据标签
X, y = load_data()

#划分数据集，test_size为测试集占总数据的比例
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=0)

其中，`X`为数据的特征，`y`为数据的标签，`test_size`表示测试集占总数据集的比例（例如`test_size=0.2`表示使用20%的数据作为测试集），`random_state`用于设定随机种子，保证每次随机划分的结果一致。 `train_test_split`方法会返回4个数组：`X_train`为训练集的特征数据，`y_train`为训练集的标签数据，`X_test`为测试集的特征数据，`y_test`为训练集的标签数据。