Multi label possible? - Githubissues

szc19990412 / TransMIL

TransMIL: Transformer based Correlated Multiple Instance Learning for Whole Slide Image Classification

341 stars 73 forks source link

Multi label possible? #49

Open Fel1xJulian opened 4 months ago

Fel1xJulian commented 4 months ago

Hi quick question,

is it possible to adapt your model to a multi label setting? The prediction layer etc. are not an issue, but would there be any issues with the attention mechanism?