Classification of Cardiotocography Based on the Apriori Algorithm and Multi-Model Ensemble Classifier

Abstract

Cardiotocography (CTG) recorded fetal heart rate and its temporal relationship with uterine contractions. CTG intelligent classification plays an important role in evaluating fetal health and protecting fetal normal growth and development throughout pregnancy. At the feature selection level, this study uses the Apriori algorithm to search frequent item sets for feature extraction. At the level of the classification model, the combination model of AdaBoost and random forest with the highest classification accuracy is finally selected by comparing various models. The suspicious class data in the CTG data set affect the overall classification accuracy. The number of suspicious class data is predicted by the multi-model ensemble method. Finally, the data set is fused from three classifications to two classifications. The classification accuracy is 0.976, and the AUC is 0.98, which significantly improves the classification effect. In conclusion, the method used in this study has high accuracy in model classification, which is helpful to improve the accuracy of fetal abnormality detection.