Machine Learning Algorithms for Transportation Mode Prediction: A Comparative Analysis

Abstract

This study investigates the performance of various machine learning (ML) algorithms in predicting transportation modes from large datasets. The investigated algorithms include Multilayer Perceptron (MLP), K-Nearest Neighbors (KNN), Decision Tree (DT), Long Short-Term Memory (LSTM), Recurrent Neural Network (RNN), and Logistic Regression. We rigorously evaluated each algorithm's performance using a robust set of metrics such as precision, recall, and F1-score. This study comprehensively explains the algorithm's capabilities, strengths, and potential weaknesses across seven transportation categories: 'walk', 'bike ', 'bus', 'car', 'taxi', 'train', and 'subway'. The DT model consistently outperformed the others, demonstrating superior accuracy and an adequate balance of precision and recall across all modes of transportation. Specifically, it achieved precision, recall, and F1 scores of around 83% to 94% across all categories. These findings underline the suitability of the DT model for this classification task and its potential for further applications in transportation mode prediction based on large datasets. However, other algorithms like LSTM and RNN also showed promising results in certain categories, suggesting the value of continued exploration of different models depending on specific use cases.Povzetek: Raziskava preučuje učinkovitost algoritmov strojnega učenja pri napovedovanju načinov prevoza iz obsežnih podatkovnih zbirk, pri čemer izstopa model odločitvenih dreves.