Use train model to predict test data also