Lyrics classification using Naive Bayes
Dalibor Bužić, Jasminka Dobša · 2018
Text classification is an important and common task in supervised machine learning. The Naive Bayes Classifier is a popular algorithm that can be used for this purpose. The goal of our research was prediction of song performer using Naive Bayes classification algorithm based solely on lyrics. A dataset that has been created consists of lyrics performed by Nirvana and Metallica, 207 songs in total. Model evaluation measures showed very good results: precision of 0.93, recall of 0.95 and F1-measure of 0.94, therefore lyrics classification using Naive Bayes can be considered as successful.