gpleiss / temperature_scaling

A simple way to calibrate your neural network.
MIT License
1.09k stars 159 forks source link

Does this trick apply for multilabel classification? #26

Open riven314 opened 3 years ago

riven314 commented 3 years ago

Great work! From the softmax setting, it seems the trick is typically applied on multi-class classification. Do you think the trick can be extended to multi-label classification? If so, what would be a possible way to extend that?