mean-squared-error code snippets in python