Better random forests

2024-04-29 15:30:39 +02:00
parent 0b8f4ddb2e
commit 0cf89d685f
8 changed files with 987 additions and 18796 deletions
--- a/data/data_loading.py
+++ b/data/data_loading.py
@@ -57,5 +57,5 @@ def load_raw_data() -> tuple[pd.DataFrame, list[pd.DataFrame]]:

 def load_data(name: str, path: os.path = os.path.join("data")) -> tuple[pd.DataFrame, pd.DataFrame]:
    metadata = pd.read_csv(os.path.join(path, name, "metadata.csv"))
-    experiments = pd.read_csv(os.path.join(path, name, "experiments.csv"))
+    experiments = pd.read_csv(os.path.join(path, name, "experiments.csv"), dtype=float)
    return metadata, experiments