Application of machine learning approaches to predict the 5-year survival status of patients with esophageal cancer.

ABSTRACT:

Background

Accurate prognostic estimation for esophageal cancer (EC) patients plays an important role in the process of clinical decision-making. The objective of this study was to develop an effective model to predict the 5-year survival status of EC patients using machine learning (ML) algorithms.

Methods

We retrieved the information of patients diagnosed with EC between 2010 and 2015 from the Surveillance, Epidemiology, and End Results (SEER) Program, including 24 features. A total of 8 ML models were applied to the selected dataset to classify the EC patients in terms of 5-year survival status, including 3 newly developed gradient boosting models (GBM), XGBoost, CatBoost, and LightGBM, 2 commonly used tree-based models, gradient boosting decision trees (GBDT) and random

SUBMITTER: Gong X 

PROVIDER: S-EPMC8662490 | biostudies-literature | 2021 Nov

REPOSITORIES: biostudies-literature