Original scientific paper
https://doi.org/10.2478/bsrj-2014-0021
A Comparison of Machine Learning Methods in a High-Dimensional Classification Problem
Marijana Zekić-Sušac
; University of Josip Juraj Strossmayer in Osijek, Faculty of Economics, Croatia
Sanja Pfeifer
orcid.org/0000-0002-7394-3080
; University of Josip Juraj Strossmayer in Osijek, Faculty of Economics, Croatia
Nataša Šarlija
orcid.org/0000-0003-2600-9735
; University of Josip Juraj Strossmayer in Osijek, Faculty of Economics, Croatia
Abstract
Background: Large-dimensional data modelling often relies on variable reduction methods in the pre-processing and in the post-processing stage. However, such a reduction usually provides less information and yields a lower accuracy of the model. Objectives: The aim of this paper is to assess the high-dimensional classification problem of recognizing entrepreneurial intentions of students by machine learning methods. Methods/Approach: Four methods were tested: artificial neural networks, CART classification trees, support vector machines, and k-nearest neighbour on the same dataset in order to compare their efficiency in the sense of classification accuracy. The performance of each method was compared on ten subsamples in a 10-fold cross-validation procedure in order to assess computing sensitivity and specificity of each model. Results: The artificial neural network model based on multilayer perceptron yielded a higher classification rate than the models produced by other methods. The pairwise t-test showed a statistical significance between the artificial neural network and the k-nearest neighbour model, while the difference among other methods was not statistically significant. Conclusions: Tested machine learning methods are able to learn fast and achieve high classification accuracy. However, further advancement can be assured by testing a few additional methodological refinements in machine learning methods.
Keywords
machine learning; support vector machines; artificial neural networks; CART classification trees; k-nearest neighbour; large-dimensional data; cross-validation
Hrčak ID:
126919
URI
Publication date:
1.9.2014.
Visits: 1.746 *