Fine tuning Roberta to classify movies into their respective genres based on their titles and synopses.
https://huggingface.co/datasets/datadrivenscience/movie-genre-prediction
The dataset contains ~100,000 movies. Each entry includes the original title, the genre and the synopsis of the movie. It is a mix of both original and AI-generated titles, genres, and synopses. The 10 genres include action, adventure, crime, family, fantasy, horror, mystery, romance, scifi, and thriller.