get_combined_anxiety_score()

src/Dataset.py

@@ -17,9 +17,23 @@ class Dataset:
             pd.DataFrame: resulting preprocessed dataframe.
         """
         dataframe = raw_dataframe.drop(["League"], axis="columns")
+        dataframe["Anxiety_score"] = self.get_combined_anxiety_score(dataframe)
         # more preprocessing goes here
         return dataframe
 
+    def get_combined_anxiety_score(self, dataframe: pd.DataFrame):
+        gad_max = 21
+        gad_min = 0
+        gad_normalised = (dataframe["GAD_T"] - gad_min) / gad_max
+        spin_max = 68
+        spin_min = 0
+        spin_normalised = (dataframe["SPIN_T"] - spin_min) / spin_max
+        swl_max = 35
+        swl_min = 5
+        swl_flipped = 1 - (dataframe["SWL_T"] - swl_min) / swl_max
+        combined = (gad_normalised + spin_normalised + swl_flipped) / 3
+        return combined
+
     def get_dataframe(self) -> pd.DataFrame:
         """A getter function for the dataframe.