Write code to train the model and evaluate it on test data.