X_train: <class 'numpy.ndarray'> (790,) y_train: <class 'numpy.ndarray'> (790,)这样两个数据集怎样修改能满足上述要求
时间: 2024-05-11 14:19:15 浏览: 133
浅谈keras通过model.fit_generator训练模型(节省内存)
为了满足上述要求,我们需要将 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(只有一个输出变量)。
阅读全文