- Book Chapter
2
- 10.1007/978-981-15-7533-4_60
Using an Ensemble Learning Approach on Traditional Machine Learning Methods to Solve a Multi-Label Classification Problem
- Jan 01, 2021
- Siddharth Basu + 3 more +3
This paper explores an interesting way to solve a multi-label classification problem using an ensemble learning with natural language processing approach. The multi-label classification problem selected is that of movie genre prediction where each movie can possibly belong to multiple genres which essentially differentiates it from a much more classical multi-class classification problem. In the current machine learning environment, there are a host of different classifiers, some which are more suited to handle more than binary classes at once such as random forests and decision trees while some which are modelled around solving an inherently binary classification problem. As we are using textual data in the form of synopsis of various movies, we use Word2Vec to vectorize the textual data in order to fit the classifiers. In this project, we use classifiers belonging to both these types, namely Naive Bayes, random forest classifier and XGBoost classifier to individually predict the multiple labels for each movie, and then we construct an ensemble model using a voting classier in the purview of ensemble learning by adjusting the weights of the combining classifiers and show that the ensemble model performs much better than each of the individual classifiers. The final micro-f-score achieved is 0.6738 which is commendable with using non-neural traditional machine learning techniques and is an improvement from the referred work for this paper.
Read more