engleski

Lyrics Classification using Naive Bayes

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.

Naive Bayes classifier, text classification, machine learning

nije evidentirano