def predict(parameters, X): A2, cache = forward_pass(X, parameters) predictions = np.round(A2) return predictions