Abstract
: Surgical resection is the standard treatment for early-stage lung cancer. Since postoperative lung function is related to mortality, predicted postoperative lung function is used to determine the treatment modality. The aim of this study was to evaluate the predictive performance of linear regression and machine learning models. We extracted data from the Clinical Data Warehouse (CDW) and developed 3 sets: set Ⅰ, the linear regression model; set Ⅱ, machine learning models omitting the missing data: and set Ⅲ, machine learning models imputing the missing data. Six machine learning models, the least absolute shrinkage and selection operator (LASSO), ridge regression, ElasticNet, Random Forest, eXtreme gradient boosting (XGBoost), and the light gradient boosting machine (LightGBM) were implemented. The forced expiratory volume in one second (FEV1) measured 6 months after surgery was defined as the outcome. Five-fold cross-validation was performed for hyperparameter tuning of the machine learning models. The dataset was split into training and test datasets at a 70:30 ratio. Implementation was done after dataset splitting in set Ⅲ. Predictive performance was evaluated by R2 and mean squared error (MSE) in the 3 sets. A total of 1,487 patients were included in sets Ⅰ and Ⅲ and 896 patients were included in set Ⅱ. In set Ⅰ, the R2 value was 0.27 and in set Ⅱ, LightGBM was the best model with the highest R2 value of 0.5 and the lowest MSE of 154.95. In set Ⅲ, LightGBM was the best model with the highest R2 value of 0.56 and the lowest MSE of 174.07. The LightGBM model showed the best performance in predicting postoperative lung function.
Talk to us
Join us for a 30 min session where you can share your feedback and ask us any queries you have
Disclaimer: All third-party content on this website/platform is and will remain the property of their respective owners and is provided on "as is" basis without any warranties, express or implied. Use of third-party content does not indicate any affiliation, sponsorship with or endorsement by them. Any references to third-party content is to identify the corresponding services and shall be considered fair use under The CopyrightLaw.