setting up the models for the medical data set