Stretch: Class Imbalance Handling

Implement:

def weighted_bce_loss(p: np.ndarray, y: np.ndarray, class_weights: dict) -> float:
    """class_weights: e.g. {0: 1.0, 1: 5.0}"""