phospho-small

This is a SetFit model that can be used for Text Classification on CPU.

The model has been trained using an efficient few-shot learning technique.

Usage

from setfit import SetFitModel

model = SetFitModel.from_pretrained("phospho-small-b72441a")

outputs = model.predict(["This is a sentence to classify", "Another sentence"])
# tensor([1, 0])

References

This work was possible thanks to the SetFit library and the work of:

Tunstall, Lewis and Reimers, Nils and Jo, Unso Eun Seo and Bates, Luke and Korat, Daniel and Wasserblat, Moshe and Pereg, Oren (2022). Efficient Few-Shot Learning Without Prompts.

ArXiv: https://doi.org/10.48550/arxiv.2209.11055

Downloads last month
9
Safetensors
Model size
109M params
Tensor type
F32
·
Inference Examples
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated) instead.