Saiftyfirst/imbalanced class weighting

jeshraghian / snntorch

Deep and online learning with spiking neural networks in Python

https://snntorch.readthedocs.io/en/latest/

MIT License

1.28k stars 217 forks source link

Saiftyfirst/imbalanced class weighting #255

Closed saiftyfirst closed 8 months ago

jeshraghian commented 8 months ago

I'm sorry this took so long, but this looks great. Merging!

jeshraghian commented 8 months ago

I'm going to keep this PR open to remind me add docstrings to the loss functions for reduction and weight parameters. Won't complain if you beat me to it though