import jax.numpy as jnp
def MSE(x, y):
return ((x - y)**2).mean()
def MSE_3D(x, y):
return ((x - y)**2).mean(axis=0)
def MSRE(x, y):
return (((x - y) / y)**2).mean()
