Evaluation of feature selection using information gain and gain ratio on bank marketing classification using naïve bayes

Open

B. Prasetiyo, Alamsyah, M.A. Muslim, N. Baroroh

2021 Journal of Physics: Conference Series Vol. 1918 Issue 4 Conference paper Cited by 25 Quartile

Abstract

One of the efforts of banks to do marketing is by telephone to offer their products, such as deposits. There are many variables that influence whether the customer decides to subscribe or not. In this study, we present a comparison of feature selection from high features dataset. We use a bank marketing dataset which has 20 features and consists of 4,119 instances. We consider 2 ranking methods entropy-based, namely Information Gain (IG) and Gain Ratio (GR). In our experiment, we classified the various selected based on the ranking of the selected features using Naïve Bayes. We show that the selection of different features is important for classification accuracy. The different combinations of feature selection can affect the accuracy results. © Published under licence by IOP Publishing Ltd.

Affiliations

Department of Computer Science, Faculty of Mathematics and Natural Sciences, Universitas Negeri Semarang, Indonesia; Department of Accounting, Faculty Economy, Universitas Negeri Semarang, Indonesia