HHHedo / IBMIL

CVPR 2023 Highlight
59 stars 9 forks source link

Num Classes #14

Closed bryanwong17 closed 6 months ago

bryanwong17 commented 6 months ago

Hi, if I want to train camelyon16 and tcga datasets (following your experiment setup), what num_classes should I set? 1 or 2? I am a little bit confused as in the DSMIL GitHub repo, its written something like this "For binary classifier, use 1 for positive bags and 0 for negative bags. Use --num_classes=1 at training."