def prob2label(prod):
    """Transforms Probability to 0/1 Labels

    Args:
        prod: Probability of prediction (confidence)

    Returns:
        int: 0/1 Labels

    """
    return (prod > 0.5)