use y_pred to predict