Mulitlabel: Clf.predict(return_proba=True) only returns probabilities for labels over the threshold

webis-de / small-text

Active Learning for Text Classification in Python

https://small-text.readthedocs.io/

MIT License

586 stars 65 forks source link

Mulitlabel: Clf.predict(return_proba=True) only returns probabilities for labels over the threshold #14

Open KimBue opened 2 years ago

KimBue commented 2 years ago

Some query strategies require the probabilities for all labels of a sample, currently only probabilities for successfully predicted labels are returned.