Search code examples
pythonarraysnumpymeanmean-square-error

Mean Squared Error in Numpy?

Is there a method in numpy for calculating the Mean Squared Error between two matrices?

I've tried searching but found none. Is it under a different name?

If there isn't, how do you overcome this? Do you write it yourself or use a different lib?

Solution

  • You can use:

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

    Or

    mse = (np.square(A - B)).mean(axis=ax)
    • with ax=0 the average is performed along the row, for each column, returning an array
    • with ax=1 the average is performed along the column, for each row, returning an array
    • with omitting the ax parameter (or setting it to ax=None) the average is performed element-wise along the array, returning a scalar value