1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
|
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",
]
|
