HOUSE_FEATURE_CSV_TITLE = "Hogwarts House"

NUMERICAL_FEATURE_CSV_TITLES = [
    "Arithmancy",
    "Astronomy",
    "Herbology",
    "Defense Against the Dark Arts",
    "Divination",
    "Muggle Studies",
    "Ancient Runes",
    "History of Magic",
    "Transfiguration",
    "Potions",
    "Care of Magical Creatures",
    "Charms",
    "Flying",
]

HOUSE_COLORS = {
    "gryffindor": "red",
    "slytherin": "green",
    "ravenclaw": "blue",
    "hufflepuff": "yellow",
}

# List of features selected to perform the logistic regression
LOGISTIC_REGRESSION_FEATURE_TITLES = [
    "Astronomy",
    "Defense Against the Dark Arts",
    "Charms",
]