metadata
tags:
- bert
- adapterhub:text-classification
- adapter-transformers
Adapter davanstrien/book-genre-classification for bert-base-cased
An adapter for the bert-base-cased model that was trained on the text-classification dataset and includes a prediction head for classification.
This adapter was created for usage with the adapter-transformers library.
Usage
First, install adapter-transformers:
pip install -U adapter-transformers
Note: adapter-transformers is a fork of transformers that acts as a drop-in replacement with adapter support. More
Now, the adapter can be loaded and activated like this:
from transformers import AutoModelWithHeads
model = AutoModelWithHeads.from_pretrained("bert-base-cased")
adapter_name = model.load_adapter("davanstrien/book-genre-classification", source="hf", set_active=True)