在机器学习模型训练中,数据需要保持一致的维度。如果您有一个 93 行 9 列的矩阵 X,而 y 是一个包含 93 个元素的一维数组,那么您需要将 y 转换为一个形状为 (93,1) 的二维数组。

可以使用 NumPy 的 reshape() 方法来实现这一点。以下代码示例演示了如何将 y 数组转换为二维数组:

import numpy as np

X = np.array([[1,2,3],[4,5,6],[7,8,9]])
y = np.array([1,0,1])

print(np.shape(X))  # 输出 (3,3)
print(np.shape(y))  # 输出 (3,)

y = y.reshape(-1, 1)

print(np.shape(y))  # 输出 (3,1)

在上述代码中,我们首先创建了一个 3 行 3 列的 X 矩阵和一个包含 3 个元素的 y 数组。然后,我们使用 reshape() 方法将 y 转换为一个形状为 (3,1) 的二维数组。-1 参数表示自动计算数组长度,根据数组中的元素数量自动计算出新的形状。

通过将 y 数组调整为与 X 矩阵相同的维度,您可以确保数据格式正确,从而顺利进行机器学习模型的训练。

NumPy 数组形状调整:将一维数组转换为二维数组

原文地址: https://www.cveoy.top/t/topic/f2ae 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录