:sparkles: Add evidential classification

ENSTA-U2IS-AI / torch-uncertainty

Open-source framework for uncertainty and deep learning models in PyTorch :seedling:

https://torch-uncertainty.github.io

Apache License 2.0

223 stars 16 forks source link

:sparkles: Add evidential classification #56

Closed xuanlongORZ closed 8 months ago

xuanlongORZ commented 8 months ago

[x] Handle the number of epochs for the simulated annealing
[x] Add the method to the ReadMe

o-laurent commented 8 months ago

Many thanks for the great PR; I took some liberty to start the modifications! Tell me if you disagree with a part of them!

o-laurent commented 8 months ago

It seems ready to merge for me. We may need to detail the loss more in the tutorial + add some more tests, but we will do this in the PR on main or just on dev! Thanks again for your help :)

Just have to fix the if/elif/else combination of the loss and handle the number of epochs in the training procedure