[`feat`] Set labels based on head classes, if possible

huggingface / setfit

Efficient few-shot learning with Sentence Transformers

https://hf.co/docs/setfit

Apache License 2.0

2.24k stars 223 forks source link

[`feat`] Set labels based on head classes, if possible #476

Closed tomaarsen closed 10 months ago

tomaarsen commented 10 months ago

Hello!

Pull Request overview

Set labels based on head classes, if possible

Details

If users train with string labels, then the (logistic regression) head will store the classes internally. We can adopt these for the SetFit labels argument, allowing these labels to be more easily accessible.

Tom Aarsen

HuggingFaceDocBuilderDev commented 10 months ago

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.