davanstrien's picture
davanstrien HF staff
add model
3953f0d
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)

Architecture & Training

Evaluation results

Citation