lvwerra / text-classification Open weights 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

224 K

Likes

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

Download history

daily snapshots · 106 days

▲ 0 in the last 30 days (0.0%)

621 K224 K

Jul 28Sep 1Oct 6Nov 10

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 14.8 M finbert 3.3 M distilbert-base-uncased-finetuned-sst-2-english 2.9 M twitter-roberta-base-sentiment-latest 2.4 M bge-reranker-base 2.4 M Prompt-Guard-86M 1.7 M