fancyzhx/amazon_polarity
Viewer • Updated • 4M • 9.25k • 51
How to use AdapterHub/xlm-roberta-base-lora-cls-amazon-polarity with Adapters:
from adapters import AutoAdapterModel
model = AutoAdapterModel.from_pretrained("google-t5/t5-base")
model.load_adapter("AdapterHub/xlm-roberta-base-lora-cls-amazon-polarity", set_active=True)lenglaender/xlm-roberta-base-lora-cls-amazon-polarity for google-t5/t5-base
An adapter for the google-t5/t5-base model that was trained on the amazon_polarity dataset and includes a prediction head for classification.
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("google-t5/t5-base")
adapter_name = model.load_adapter("AdapterHub/xlm-roberta-base-lora-cls-amazon-polarity", source="hf", set_active=True)
LoRA has r=8 and alpha=8 and was trained with dropout=0.1
Accuracy on Amazon polarity dataset: 95.91%