sa-

# Compute ndcg @ k

def ndcg_at_k(predictions_df, k):
    """
    This pandas dataframe should contain the columns "customer_id",
    "estimate", and "label". 
    
    Where `estimate` is a recommendation score
    that we can sort by descending order.