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

337 K

Likes

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

Download history

daily snapshots · 55 days

▲ 146 K in the last 30 days (30.3%)

621 K225 K

Jul 28Aug 15Sep 2Sep 20

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 any text-classification model

Popular comparisons

bge-reranker-v2-m3 17.6 M finbert 5.3 M Prompt-Guard-86M 4.4 M bge-reranker-base 4.0 M distilbert-base-uncased-finetuned-sst-2-english 3.7 M twitter-roberta-base-sentiment-latest 3.2 M