Traceback (most recent call last): File "E:\PMSM\Test1\t-sne.py", line 294, in <module> y_hat = np.squeeze(y_hat,

动态 未结 0 92
小小草
小小草 LV1 2023年9月23日 23:40 发表
这个错误的原因是y_hat的第二个维度大小不为1,无法通过`np.squeeze()`函数去掉。可能是因为在训练模型时,设置的n_predictions参数过大,导致y_hat的第二个维度大小超过了1。 解决方法可以尝试调整n_predictions参数的值,或者使用其他方法去掉y_hat的第二个维度。比如,可以使用`np.reshape()`函数将y_hat的形状改变为与y_test相同的形状。具体代码如下: ```python y_hat = np.reshape(y_hat, (y_hat.shape[0], y_hat.shape[-1])) print(y_test.shape, y_hat.shape) ``` 这样就可以得到与y_test相同的形状,并且最后一列的数值不变。
收藏(0)  分享
相关标签: 智能问答
问题没解决?让AI助手帮你作答 AI助手
0个回复
  • 消灭零回复