bert-sentiment-imdb / README.md
zavora's picture
Update README.md
d5fd2be verified
metadata
license: mit
base_model:
  - google-bert/bert-base-uncased
library_name: adapter-transformers
datasets:
  - stanfordnlp/imdb
pipeline_tag: text-classification

bert-sentiment-imdb

This is a fine-tuned BERT model for sentiment analysis on the IMDB dataset.

  • Base Model: bert-base-uncased
  • Dataset: IMDB
  • Task: Sentiment Analysis (Binary)
  • Usage:
    from transformers import pipeline
    classifier = pipeline("sentiment-analysis", model="zavora/bert-sentiment-imdb")