Add probabilistic classification to hiclass

scikit-learn-contrib / hiclass

A python library for hierarchical classification compatible with scikit-learn

BSD 3-Clause "New" or "Revised" License

113 stars 20 forks source link

Add probabilistic classification to hiclass #119

Open LukasDrews97 opened 4 months ago

LukasDrews97 commented 4 months ago

Add probabilistic classification via calibration to hiclass using the following methods:

Platt Scaling
Isotonic Regression
Beta calibration
(Inductive/Cross) Venn-ABERS calibration

mirand863 commented 4 months ago

Hi @LukasDrews97,

Just a quick request from someone from France that reached out to me via e-mail. Would it be possible to add a threshold to remove labels that have low probability?