In [16]:
def predict(parameters, X, activation):
    
    y_predictions, cache = L_model_forward(X, parameters, activation)
    predictions = np.round(y_predictions)
    
    return predictions