predicting test set