Anime or Not?

An NLP model that determines whether a plot is anime enough.

Play with it on HuggingFace Space

Data Collection

The model is trained on plots that have been labelled as anime or not. I scraped IMDb lists for anime and vice-versa and then extracted plot details using IMDbPY (now Cinemagoer).

For steps to create the dataset check out the dataset notebook in the repo or open in Colab.

Dataset hosted on HuggingFace

Modelling

HuggingFace Transformers' DistilBERT, a pre-trained transformer model, is fine-tuned on the dataset of plots labelled as anime or not.

For steps to make the dataset check out the model notebook in the repo or open in Colab.

Model hosted on HuggingFace

Contributing

If you want to contribute code, simply create a pull request and the developers will review it. If you have an idea, create an issue and the developers will look into it!