y_test:(3849, 1) y_hat:(3849, 25, 1)怎么把y_hat变成与y_test一样的维度,并且最后一列的数值不变(其实就是直接把第2维度去掉)

动态 未结 0 90
小小草
小小草 LV1 2023年9月23日 23:40 发表
可以使用`numpy.squeeze()`函数将y_hat的第二个维度去掉,从而得到与y_test相同的维度。具体代码如下: ```python y_hat = np.squeeze(y_hat, axis=1) # 去掉第二个维度 print(y_test.shape, y_hat.shape) ``` 这样就可以得到与y_test相同的维度,并且最后一列的数值不变。
收藏(0)  分享
相关标签: 智能问答
问题没解决?让AI助手帮你作答 AI助手
0个回复
  • 消灭零回复