actual = [0, 1, 2, 0, 3] predicted = [0.1, 1.3, 2.1, 0.5, 3.1] mse = sklearn.metrics.mean_squared_error(actual, predicted) rmse = math.sqrt(mse) print(rmse)