Predicting postoperative liver cancer death outcomes with machine learning
Yong Wang, Chaopeng Ji, Ying Wang, Mu‐Huo Ji, Jianjun Yang, Cheng-Mao Zhou
Abstract
OBJECTIVE: To investigate the effect of 5 machine learning algorithms in predicting total hepatocellular carcinoma (HCC) postoperative death outcomes. METHODS: This study was a secondary analysis. A prognosis model was established using machine learning with python. RESULTS: The results from the machine learning gbm algorithm showed that the most important factors, ranked from first to fifth, were: preoperative aspartate aminotransferase (GOT), preoperative AFP, preoperative cereal third transaminase (GPT), preoperative total bilirubin, and LC3. Postoperative death model results for liver cancer patients in the test group: of the 5 algorithm models, the highest accuracy rate was that of forest (0.739), followed by the gbm algorithm (0.714); of the 5 algorithms, the AUC values, from high to low, were forest (0.803), GradientBoosting (0.746), gbm (0.724), Logistic (0.660) and DecisionTree (0.578). CONCLUSION: Machine learning can predict total hepatocellular carcinoma postoperative death outcomes.