For determining the Correlation between lung cancer patients who have undergone lobectomy and have a mental illness, the team developed multiple machine learning models. For this study, we split the data into 80% training data (4464 sample data) and 20% test data (1117 sample data).
We divided this problem statement into two areas of evaluation:
- Predicting LOS of a patient with both lung cancer and mental illness using only Diagnosis codes.
- Predicting LOS of a patient with both lung cancer and mental illness using both Diagnosis codes and Socio-demographic features.
The following algorithms were then developed:
- SGDRegressor
- GradientBoostingRegressor
- LinearRegression
- KNeighborsRegressor
- RandomForestRegressor
- SVR
- TensorFlow