cgarciae / treex

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

Add more metrics #41

Open anvelezec opened 2 years ago

anvelezec commented 2 years ago

The idea is to add more metrics taking as basis torchmetrics and tf metrics implementation.

Guidelines

Regression metrics

Classification metrics

If you want to work on a metric, feel free to "claim" it via a comment here so others know you are working on it. Additionally feel free to add as many metrics as desired, even from other tasks such as audio, language, image, etc.

jmarrietar commented 2 years ago

I will take Mean squared error. I will post here any or in discord the challenges or doubts I face.