Model Card for electra-movie-genre

Fine-tuned version of "deepset/electra-base-squad2" on a corpus of IMDB movie descriptions and the relevant genre.

Table of Contents

• Model Card for DistilGutenMystery
• Table of Contents
• Table of Contents
• Model Details
  • Model Description
• Uses
  • Direct Use
  • Downstream Use [Optional]
  • Out-of-Scope Use
• Bias, Risks, and Limitations
  • Recommendations
• Training Details
  • Training Data
  • Training Procedure
    • Preprocessing
    • Speeds, Sizes, Times
• Evaluation
  • Testing Data, Factors & Metrics
    • Testing Data
    • Factors
    • Metrics
  • Results
• Model Card Authors [optional]
• Model Card Contact
• How to Get Started with the Model

Model Details

Model Description

Fine-tuned version of "deepset/electra-base-squad2" on a corpus of IMDB movie descriptions and the relevant genre.

• Developed by: More information needed
• Shared by [Optional]: More information needed
• Model type: Language model
• Language(s) (NLP): en
• License: cc-by-4.0
• Parent Model: More information needed
• Resources for more information: More information needed

Uses

Direct Use

Classifying movie descriptions into 1 of 16 genres. Many movies involve multiple genres but this model attempts to classify the primary/first genre mentioned on IMDB. Genres include: Fantasy, Romance, Thriller, Biography, Horror, Action, Crime, Animation, Adventure, Mystery, War, Family, History, Scifi, Film-noir, Sports

Downstream Use

More information needed

Out-of-Scope Use

More information needed

Bias, Risks, and Limitations

More information needed

Recommendations

More information needed

Training Details

Training Data

Information regarding the training dataset can be found here: https://huggingface.co/datasets/jquigl/imdb-genres

Training Procedure

Preprocessing

More information needed

Evaluation

Testing Data, Factors & Metrics

Testing Data

More information needed

Factors

More information needed

Metrics

More information needed

Results

More information needed

Model Card Authors [optional]

Hugging Face, Jack Quigley

Model Card Contact

More information needed

How to Get Started with the Model

Use the code below to get started with the model.

Click to expand

from transformers import AutoTokenizer, pipeline
from transformers import ElectraConfig, ElectraForSequenceClassification

tokenizer = AutoTokenizer.from_pretrained("deepset/electra-base-squad2")
model = ElectraForSequenceClassification.from_pretrained('jquigl/electra-movie-genre')

classifier = pipeline("text-classification", model = model, tokenizer = tokenizer)
example = classifier("A Manny Velazquez underground slasher tribute about a number of serial killers on the loose in Chicago and begins a murder spree.")
print(example[0]) 

#LABEL_0 = Fantasy
#LABEL_1 = Romance
#LABEL_2 = Thriller
#LABEL_3 = Biography
#LABEL_4 = Horror
#LABEL_5 = Action
#LABEL_6 = Crime
#LABEL_7 = Animation
#LABEL_8 = Adventure
#LABEL_9 = Mystery
#LABEL_10 = War
#LABEL_11 = Family
#LABEL_12 = History
#LABEL_13 = Scifi
#LABEL_14 = Film-noir
#LABEL_15 = Sports