Mean Squared Error in Numpy?

Question

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?

Answer 1

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 ax=None the average is performed element-wise along the array, returning a scalar value

Answer 2

This isn't part of numpy, but it will work with numpy.ndarray objects. A numpy.matrix can be converted to a numpy.ndarray and a numpy.ndarray can be converted to a numpy.matrix.

from sklearn.metrics import mean_squared_error mse = mean_squared_error(A, B) 

See Scikit Learn mean_squared_error for documentation on how to control axis.