huggingface / setfit

Efficient few-shot learning with Sentence Transformers
https://hf.co/docs/setfit
Apache License 2.0
2.24k stars 222 forks source link

Custom sentence pairs sampling #536

Open mariusmercier opened 5 months ago

mariusmercier commented 5 months ago

Hello,

Is there a way of inputting directly a custom pair sampling?

My model takes too much time when doing "oversampling", so I am using "undersampling", however I have specific negative pairs that I would like to include. For each category I have a text that is the label of the category, so I would like that each labels are included in the negative pairs.

Thanks !