search

cgarciae / treex

A Pytree Module system for Deep Learning in JAX
https://cgarciae.github.io/treex/
MIT License
214 stars 17 forks source link

Add MSE and tests #50

Closed jmarrietar closed 2 years ago

jmarrietar commented 2 years ago

Mean Square Error (MSE) metric and tests using as Reference MAE implementation at PR (mostly the same code)

cgarciae commented 2 years ago

Thanks @jmarrietar ! LGTM Merging...