Trusted answers to developer questions

Calculating Mean Squared Error in Python

Get the Learn to Code Starter Pack

Break into tech with the logic & computer science skills you’d learn in a bootcamp or university — at a fraction of the cost. Educative's hand-on curriculum is perfect for new learners hoping to launch a career.

Defining the variables

$n$ - the total number of terms for which the error is to be calculated
$y_i$ - the observed value of the variable
$\bar y_i$ - the predicted value of the variable

The mean square error is the average of the square of the difference between the observed and predicted values of a variable.

In Python, the MSE can be calculated rather easily, especially with the use of lists.

How to calculate MSE

Calculate the difference between each pair of the observed and predicted value
Take the square of the difference value
Add each of the squared differences to find the cumulative values
In order to obtain the average value, divide the cumulative value by the total number of items in the list

Example

Suppose you wish to calculate the MSE and are provided with the observed and predicted values. The steps mentioned above will be implemented as follows:

y = [11,20,19,17,10]
y_bar = [12,18,19.5,18,9]
summation = 0  #variable to store the summation of differences
n = len(y) #finding total number of items in list
for i in range (0,n):  #looping through each element of the list
  difference = y[i] - y_bar[i]  #finding the difference between observed and predicted value
  squared_difference = difference**2  #taking square of the differene 
  summation = summation + squared_difference  #taking a sum of all the differences
MSE = summation/n  #dividing summation by total values to obtain average
print "The Mean Square Error is: " , MSE

RELATED TAGS

error

python

mean square error

mse

Learn in-demand tech skills in half the time

PRODUCTS

Mock Interview

New

Courses

Skill Paths

Projects

Assessments