Setting up machine learning models to be applied to the test set in order to predict survival.