Hugging Face's logo

AdapterHub
/
gpt2_sentiment_sst-2_pfeiffer

Text Classification
Adapters
gpt2
adapterhub:sentiment/sst-2
Model card Files Community
gpt2_sentiment_sst-2_pfeiffer / README.md
calpt's picture
calpt
Add adapter gpt2_sentiment_sst-2_pfeiffer version 1
c961366 verified
preview code
|
raw
history blame
1.24 kB
metadata
tags:
  - gpt2
  - text-classification
  - adapter-transformers
  - adapterhub:sentiment/sst-2
license: apache-2.0

Adapter gpt2_sentiment_sst-2_pfeiffer for gpt2

Adapter for gpt2 in Pfeiffer architecture trained on the SST-2 dataset for 10 epochs with 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("gpt2")
adapter_name = model.load_adapter("AdapterHub/gpt2_sentiment_sst-2_pfeiffer")
model.set_active_adapters(adapter_name)

Architecture & Training

  • Adapter architecture: pfeiffer
  • Prediction head: classification
  • Dataset: SST-2

Author Information

Citation

This adapter has been auto-imported from https://github.com/Adapter-Hub/Hub/blob/master/adapters/ukp/gpt2_sentiment_sst-2_pfeiffer.yaml.