Predicting a failure of postoperative thromboprophylaxis in non-small cell lung cancer: A stacking machine learning
Ligang Hao1, Junjie Zhang2, Yonghui Di1
1Department of Thoracic Surgery, Xingtai People's Hospital, Xingtai, Hebei, China.
Background:
Non-small-cell lung cancer (NSCLC) and its surgery significantly increase the venous thromboembolism (VTE) risk. This study explored the VTE risk factors and established a machine-learning model to predict a failure of postoperative thromboprophylaxis.
Methods:
This retrospective study included patients with NSCLC who underwent surgery between January 2018 and November 2022. The patients were randomized 7:3 to the training and test sets. Nine machine learning models were constructed. The three most predictive machine-learning classifiers were chosen as the first layer of the stacking machine-learning model, and logistic regression was the second layer of the meta-learning model.
Results:
This study included 362 patients, including 58 (16.0%) with VTE. Based on the multivariable logistic regression analysis, age, platelets, D-dimers, albumin, smoking history, and epidermal growth factor receptor (EGFR) exon 21 mutation were used to develop the nine machine-learning models. LGBM Classifier, RandomForest Classifier, and GNB were chosen for the first layer of the stacking machine learning model. The area under the received operating characteristics curve (ROC-AUC), accuracy, sensitivity, and specificity of the stacking machine learning model in the training/test set were 0.984/0.979, 0.949/0.954, 0.935/1.000, and 0.958/0.887, respectively. In the validation set, the final stacking machine learning model demonstrated an ROC AUC of 0.983, accuracy of 0.937, sensitivity of 0.978, and specificity of 0.947. The decision curve analyses revealed high benefits.
Conclusion:
The stacking machine learning model based on EGFR mutation and clinical characteristics had a predictive value for postoperative VTE in patients with NSCLC.
