predictions for test data and figures