paper preperations and notebooks, optuna callbacks

2021-04-02 08:45:11 +02:00
parent 7c88602776
commit cec3a07d60
21 changed files with 3818 additions and 1059 deletions
--- a/datasets/compare_base.py
+++ b/datasets/compare_base.py
@@ -118,7 +118,9 @@ class CompareBase(_BaseDataModule):
                lab_file = None

        for data_option in data_options:
-            if lab_file is not None:
+            if lab_file is None:
+                lab_file = f'{data_option}.csv'
+            elif lab_file is not None:
                if any([x in lab_file for x in data_options]):
                    lab_file = f'{data_option}.csv'
            dataset = self._load_from_file(lab_file, data_option, rebuild=True)