lvwerra / text-classification updated 3 years ago

distilbert-imdb

Name: distilbert-imdb
Author: lvwerra

This model is a fine-tuned version of distilbert-base-uncased on the imdb dataset (training notebook is here). It achieves the following results on the evaluation set: - Loss: 0.1903 - Accuracy: 0.928

Params

—

Context

512

Downloads 30d

261 K

Likes

Commercial use: allowed apache-2.0 Not gated View on Hugging Face ↗

Download history

daily snapshots · 10 days

262 K225 K

Jul 28Jul 31Aug 3Aug 6

Specifications

Architecture: DistilBertForSequenceClassification
Context length: 512
Vocabulary: 30,522
Licence: apache-2.0
First seen on the Hub: 2022-03-02
Training datasets: imdb
imdb (reported): 0.928
Added to our catalog: 2026-07-28

Compare with

bge-reranker-v2-m3 19.2 M finbert 5.6 M bge-reranker-base 4.6 M Prompt-Guard-86M 4.5 M distilbert-base-uncased-finetuned-sst-2-english 3.7 M koelectra-small-v3-nsmc 2.9 M

Appears in

Best open coding models Commercial-use text models Models that fit in 12 GB