X_train: <class 'numpy.ndarray'> (790,) y_train: <class 'numpy.ndarray'> (790,)这样两个数据集怎样修改能满足上述要求
时间: 2024-05-11 15:19:15 浏览: 130
train数据集
5星 · 资源好评率100%
为了满足上述要求,我们需要将 X_train 和 y_train 转换为二维数组,其中 X_train 的第一维表示样本数,第二维表示特征数;y_train 的第一维表示样本数,只有一个输出变量。
可以使用 reshape() 函数将 X_train 和 y_train 转换为二维数组:
```python
X_train = X_train.reshape(-1, 1)
y_train = y_train.reshape(-1, 1)
```
这将把 X_train 和 y_train 转换为二维数组,其中 X_train 的第一维度为样本数,第二维度为 1(只有一个特征),y_train 的第一维度为样本数,第二维度也为 1(只有一个输出变量)。
阅读全文