Using machine learning models to better predict bladder cancer stages