search

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

multitask MIL #48

Closed M0hammadL closed 2 years ago

M0hammadL commented 2 years ago

Adding ability to add multiple MIL heads with their attention scores and they could either classification or regression.

alitinet commented 2 years ago

@M0hammadL shall we have parameters like classification, regression and ordinal_regression when initializing the model to specify this? and then the user can specify a list of columns in .obs to use for each of those i.e. ordinal_regression = ['severity'], regression=['age'] and another parameter to specify the ordinal order i.e. ordinal_regression_order = [['healthy', 'mild', 'severe']]

M0hammadL commented 2 years ago

yes, that sounds great