huggingface / huggingface/setfit

Specifying ones own samples or sentence pairs

Open
#576 0 comments 0 reactions 0 assignees View on GitHub
Dominant language
Jupyter Notebook
Stars
2.8k
Forks
267
Avg merge
36m
Merged PRs (30d)
5

Description

I have been looking into SetFit for a while and its a very nice tool. Currently, it seems that there are a lot of options for training and using the classification head, but there is no way to train the 'body' on a specified set of label pairs. Would it be possible or appreciated if such capacity was added?

One of the key reasons is that currently all samples/classes are treated the same when doing contrastive learning (correct me if wrong). It would be nice to specify the pairs myself as to enhance the contrastive capacity (hard negatives / positives).

If there are already options for this let me know.

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.