guilhem_lavaux/JaxPM
0
0
Fork 1
mirror of https://github.com/DifferentiableUniverseInitiative/JaxPM.git synced 2025-05-15 04:21:12 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Format

Browse source
This commit is contained in:
Wassim Kabalan 2024-12-08 22:45:09 +01:00
parent 7823fdaf98
commit af29c4005d
7 changed files with 68 additions and 63 deletions
Download patch file Download diff file Expand all files Collapse all files

11
tests/helpers.py
View file

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