Abstract
Uterine cancer prediction accuracy is important in clinical decision-making because it improves the overall chances of patient recovery. Several machine learning models, such as Decision Tree, Random Forest, XGBoost Regressor, and Support Vector Regressor, were explored to determine which is more effective in predicting uterine cancer. Attributes such as mutation counts, diagnosis age, and MSI score, were used for the analysis. The different models were tested using the standard performance metrics such as the Mean Absolute Error (MAE), Root Mean Squared Error (RMSE), and R2 Score. Random Forest showed the highest predictive performance with an R2 score of 0.655, followed by XGBoost regressor, which was relatively close to the R2 score of Random Forest. Support Vector Regressor performed very poorly as the R2 score was negative, implying that the model is not suitable for such prediction. Ensemble-based models, which include Random Forest and XGBoost Regressor, have proven to be more effective in handling medical prediction tasks, and this is because of their robustness and their ability when it comes to handle overfitting. Though model generalizability was affected due to small data size and the absence of hyperparameter tuning. The future work will focus on expanding the dataset, implementing hyperparameter tuning, integrating deep learning, and leveraging explainable AI (XAI). The research has provided valuable insight for clinicians who wish to use machine learning for uterine cancer prognosis.