def one_hot_encode(label):
    """
    Given a label - "red", "green", or "yellow". Returns a one-hot encoded label
    """
    if label == "red":
        return [1, 0, 0]
    if label == "green":
        return [0, 0, 1]
    return [0, 1, 0]