1.Research on gastric cancer lymph node metastasis prediction model based on machine learning algorithms
Haomin SHI ; Su YAN ; Mengmeng QIAO ; Huilian YANG
Journal of Clinical Medicine in Practice 2024;28(1):41-47
Objective To establish and validate a prediction model for gastric cancer lymph node metastasis based on four machine learning (ML) algorithms. Methods A retrospective analysis was conducted on clinical data of 531 patients who underwent radical gastrectomy. The patients were randomly divided into training set (399 patients) and test set (132 patients) in a ratio of 3 to 1. Univariate analysis was used to screen for variables associated with gastric cancer lymph node metastasis, and Logistic regression, random forest, K-nearest neighbor algorithm, and support vector machine algorithm models were established to rank the importance of variables. All ML algorithm models were validated in the test set, and receiver operating characteristic (ROC) curves were plotted. The optimal ML algorithm model was determined based on the area under the curve (