theislab / multimil

Multimodal weakly supervised learning to identify disease-specific changes in single-cell atlases
https://multimil.rtfd.io/
BSD 3-Clause "New" or "Revised" License
19 stars 3 forks source link

add ordinal classification #49

Closed M0hammadL closed 2 years ago

M0hammadL commented 2 years ago

as discussed, let's have as an option to provide a dictionary order = {"class_x":0, " class_y":1, ...} and if provided we can make it into regression with mse loss instead