HE Long-ping, YAO Nan, WANG Qi-hu, et al. Rock burst intensity grading prediction model based on
automatic machine learning[J]. 2024, 45(9): 2839-2848.
DOI:
HE Long-ping, YAO Nan, WANG Qi-hu, et al. Rock burst intensity grading prediction model based on
automatic machine learning[J]. 2024, 45(9): 2839-2848.DOI: 10.16285/j.rsm.2023.1607.
Rock burst intensity grading prediction model based on
automatic machine learning
To address issues related to excessive human influence and prolonged prediction times in rockburst prediction
we propose a rockburst intensity classification prediction model based on automatic machine learning. This model is trained using five automatic machine learning frameworks and evaluated using metrics such as accuracy
precision
recall
and F1-score. Subsequently
we compare the performance of this trained model with results from thirteen common machine learning models. The model developed with the Auto-Sklearn framework achieved a high accuracy of 0.969
while the model created with the Auto-Gluon framework
although having the lowest accuracy among the five frameworks
still achieved an accuracy of 0.927. Rockburst prediction models constructed using AutoML frameworks significantly outperformed traditional machine learning algorithms. The Auto-Sklearn-based model exhibited the highest accuracy. In summary
the optimized model was applied to predict rockburst events at the Shaiqi River phosphate mine
and the predictions were consistent with the actual observations on-site. This indicates that the automatic machine learning-based model for rockburst intensity classification prediction can accurately predict rockburst incidents in real-world engineering settings.