Downloads · 30 days
6
32% of all-time downloads
facebook-bart-base_sts_sts-b_pfeiffer
AdapterHub/facebook-bart-base_sts_sts-b_pfeiffer
facebook-bart-base_sts_sts-b_pfeiffer is a text classification model from AdapterHub. Use it when you need a label for a piece of text. It is set up for adapter-transformers. The card lists the license as apache-2.0.
Adapter for bart-base in Pfeiffer architecture trained on the STS-B dataset for 15 epochs with early stopping and a learning rate of 1e-4.
All-time downloads
19
Public
Repo size
6 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin6 MB · 100%
At a glance
- Task
- Text Classification
- Library
- adapter-transformers
- License
- apache-2.0
- Access
- Public
- Created
- May 5, 2024
- Updated
- May 5, 2024
- SHA
- 926a9bd8
Try a prompt
Model card
From the Hugging Face model README
Adapter facebook-bart-base_sts_sts-b_pfeiffer for facebook/bart-base
Adapter for bart-base in Pfeiffer architecture trained on the STS-B dataset for 15 epochs with early stopping and a learning rate of 1e-4.
This adapter was created for usage with the Adapters library.
Usage
First, install adapters:
pip install -U adapters
Now, the adapter can be loaded and activated like this:
from adapters import AutoAdapterModel
model = AutoAdapterModel.from_pretrained("facebook/bart-base")
adapter_name = model.load_adapter("AdapterHub/facebook-bart-base_sts_sts-b_pfeiffer")
model.set_active_adapters(adapter_name)
Architecture & Training
- Adapter architecture: pfeiffer
- Prediction head: classification
- Dataset: STS-B
Author Information
- Author name(s): Clifton Poth
- Author email: [email protected]
- Author links: Website, GitHub, Twitter
Citation
This adapter has been auto-imported from https://github.com/Adapter-Hub/Hub/blob/master/adapters/ukp/facebook-bart-base_sts_sts-b_pfeiffer.yaml.
- Task
- Text Classification
- Library
- adapter-transformers
- License
- apache-2.0
- Created
- May 5, 2024
- Updated
- May 5, 2024