python - Numpy中的均方误差？

Question

numpy 中是否有计算两个矩阵之间的均方误差的方法？

我尝试过搜索，但没有找到。是不是用了不同的名字？

如果没有，你如何克服这个问题？你是自己写的还是使用不同的库？

Answer 1

你可以使用:

mse = ((A - B)**2).mean(axis=ax)

或者

mse = (np.square(A - B)).mean(axis=ax)

• 使用 ax=0 沿行执行平均，对于每一列，返回一个数组
• 使用 ax=1 沿列执行平均，对于每一行，返回一个数组
• 省略 ax 参数(或将其设置为 ax=None)沿数组按元素执行平均，返回一个标量值