Leveraging Machine Learning As a Prerequisite for Effective Maintenance
摘要
For the purpose of this study, a machine learning (ML) approach is used to address predictive maintenance tasks based on machine failures, facilitating effective maintenance interventions. To develop a machine failure prediction model based on a classification problem, a synthetic dataset containing 10.000 instances with 14 features is employed. Ten ML algorithms are tested: AdaBoost, Bagging Classifier, CatBoost, Decision Tree (DT), K-Nearest Neighbor (KNN), Naïve Bayes (NB), Random Forest (RF), Support Vector Classifier (SVC), Stacking Classifier, and Voting Classifier. The appropriate procedures are implemented in the Python programming language using its libraries, following a methodology conducted through proper activities, with the aim of obtaining a certain number of predicted classes. The 40 classification models are trained on all data, failure cases, the balanced dataset, and the binary balanced dataset. The results, in the form of metrics such as accuracy, precision, recall, and F1 score, are calculated and evaluated according to the criterion of the highest accuracy values. These values are cross-referenced with the confusion matrices, as well as with the F1-scores, if necessary. For all datasets, the CatBoost model proved to be the best, with accuracy rates ranging from 90.52% to 97.90%. To achieve better performance, it is necessary to further expand the dataset by including additional failure cases.