Code for training all the models used in Task-2 Prediction. 
