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
305 stars 20 forks source link

:sparkles: Enable regression & misc #27

Closed o-laurent closed 1 year ago

o-laurent commented 1 year ago

Done for now: Add an arg to CLI to split regression and cls :hammer: Modify tests accordingly :heavy_check_mark: Add a regression recipe :sparkles: Add a small MLP model :sparkles: Add an MLP baseline :sparkles: Add a UCIRegression Datamodule :sparkles: Add GaussianNLL lightning metric :sparkles: Add an experiment :sparkles:

o-laurent commented 1 year ago

Documentation is missing but the code seems approx fine. I'm working on the doc now.