Model Training

Test multiple models and find the best one (next step might help).