Prevent sampling 2x more than requested when `max_steps` is set

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

Prevent sampling 2x more than requested when `max_steps` is set #556

Closed tomaarsen closed 2 months ago

tomaarsen commented 2 months ago

Hello!

Pull Request overview

Prevent sampling 2x more than requested

Details

max_pairs was used to limit pos_pairs and neg_pairs, but that results in 2 * max_pairs being created. This fixes that by placing a // 2 over max_pairs.

Tom Aarsen

HuggingFaceDocBuilderDev commented 2 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.