1 files changed, 31 insertions, 0 deletions
diff --git a/module/constants.py b/module/constants.py
new file mode 100644
index 0000000..accb4c7
--- /dev/null
+++ b/module/constants.py
@@ -0,0 +1,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",
+]