Movie Genre Prediction Using AI-Generated Data and Natural Language Processing
Sifat Tanvir, Ayesha Parpia Abontee, Md. Abu Rayhan, Md. Fuadur Rahman, Tasnim Sultana, Ajwa Ahmed · 2023
Every movie belongs to one or more specific genre. In this work, a dataset was prepared consisting of AI-generated movie titles and plots of 10 different genres. Text vectorization was used to convert every single word of the titles and plots into vectors. These vectors were used to predict which genre each movie belongs to. For training, a sequential neural network model was built and used, and an accuracy of almost 80% was achieved.