Titanic survival prediction based on machine learning.

Authors

  • Siyu Wu

DOI:

https://doi.org/10.61173/9rh9s369

Keywords:

machine learning, Titanic guest data, predict data, model analysis

Abstract

Using the Titanic guest data offered by Kaggle, we executed data cleansing, feature engineering, and training of various designs, including logistic regression, Random Woodland, XGBoost, and LightGBM versions. We likewise brought out hyperparameter tuning and version combination, and ultimately enhanced the forecast efficiency of the version by heavy average approach. The evaluation results of the model showed that the integrated design surpassed the single design on the test collection, with greater accuracy and ROC AUC ratings.

References

[1] Emmanuel, T., Maupong, T., Mpoeleng, D., et al. “A Survey on Missing Data in Machine Learning.” Journal of Big Data, vol. 8, no. 140, 2021, https://doi.org/10.1186/s40537-021-00516- 9.

[2] Palanivinayagam, A., and R. Damaševičius. “Effective Handling of Missing Values in Datasets for Classification Using Machine Learning Methods.” Information, vol. 14, no. 2, 2023, p. 92, https://doi.org/10.3390/info14020092.

[3] Pudjihartono, N., Fadason, T., Kempa-Liehr, A. W., and J. M. O’Sullivan. “A Review of Feature Selection Methods for Machine Learning-Based Disease Risk Prediction.” Frontiers in Bioinformatics, vol. 2, 2022, p. 927312, https://doi.org/10.3389/ fbinf.2022.927312.

[4] “Discover Feature Engineering: How to Engineer Features and How to Get Good at It.” Machine Learning Mastery, https:// machinelearningmastery.com/discover-feature-engineering-howto-engineer-features-and-how-to-get-good-at-it/. Accessed 26 May 2024.

[5] Couronne, R., Probst, P., and A. L. Boulesteix. “Random Forest versus Logistic Regression: A Large-Scale Benchmark Experiment.” BMC Bioinformatics, vol. 19, no. 270, 2018, https://doi.org/10.1186/s12859-018-2264-5.

[6] Shao, Z., Ahmad, M. N., and A. Javed. “Comparison of Random Forest and XGBoost Classifiers Using Integrated Optical and SAR Features for Mapping Urban Impervious Surface.” Remote Sensing, vol. 16, no. 4, 2024, p. 665, https:// doi.org/10.3390/rs16040665.

[7] Hong, W., Zhou, X., Jin, S., Lu, Y., Pan, J., Lin, Q., Yang, S., et al. “A Comparison of XGBoost, Random Forest, and Nomograph for the Prediction of Disease Severity in Patients With COVID-19 Pneumonia: Implications of Cytokine and Immune Cell Profile.” Frontiers in Cellular and Infection Microbiology, vol. 12, 2022, p. 819267, https://doi.org/10.3389/ fcimb.2022.819267.

[8] Zhang, Y., Liu, J., and W. Shen. “A Review of Ensemble Learning Algorithms Used in Remote Sensing Applications.” Applied Sciences, vol. 12, no. 17, 2022, p. 8654, https://doi. org/10.3390/app12178654.

Downloads

Published

2024-08-14