cornell-movie-review-data/rotten_tomatoes
Viewer • Updated • 10.7k • 79.1k • 112
How to use AdapterHub/distilbert-base-uncased_sentiment_rotten_tomatoes_pfeiffer with Adapters:
from adapters import AutoAdapterModel
model = AutoAdapterModel.from_pretrained("distilbert-base-uncased")
model.load_adapter("AdapterHub/distilbert-base-uncased_sentiment_rotten_tomatoes_pfeiffer", set_active=True)distilbert-base-uncased_sentiment_rotten_tomatoes_pfeiffer for distilbert-base-uncased
Adapter for distilbert-base-uncased in Pfeiffer architecture trained on the Rotten Tomatoes dataset for 15 epochs with early stopping and a learning rate of 1e-4.
This adapter was created for usage with the Adapters library.
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("distilbert-base-uncased")
adapter_name = model.load_adapter("AdapterHub/distilbert-base-uncased_sentiment_rotten_tomatoes_pfeiffer")
model.set_active_adapters(adapter_name)
This adapter has been auto-imported from https://github.com/Adapter-Hub/Hub/blob/master/adapters/ukp/distilbert-base-uncased_sentiment_rotten_tomatoes_pfeiffer.yaml.