Update in preprocessing.py, renames the columns. Only fixed for one dataset format, if the dataset has a different format it could lead to errors again

Author: Max

lkauto/optimization_strategies/bayesian_optimization.py

@@ -120,7 +120,6 @@ def bayesian_optimization(train: Dataset,
     if cs is None:
         logger.debug('initializing default ConfigurationSpace')
         cs = get_default_configuration_space(data=train,
-                                             val_fold_indices=evaler.train_test_splits,
                                              validation=validation,
                                              feedback='explicit',
                                              random_state=random_state)

lkauto/preprocessing/preprocessing.py

@@ -65,7 +65,7 @@ def preprocess_data(data: Dataset, #data: pd.DataFrame
             data = data[[user_col, item_col, timestamp_col]]
             data.columns = ['user', 'item', 'timestamp']
         else:
-            data = data[[user_col, item_col, rating_col, timestamp_col]]
+            # data = data[[user_col, item_col, rating_col, timestamp_col]]
             data.columns = ['user', 'item', 'rating', 'timestamp']
     else:
         if rating_col is None:

lkauto/utils/get_default_configuration_space.py

@@ -16,8 +16,8 @@
 
 
 def get_default_configuration_space(data: Union[Dataset, Iterator[TTSplit]],
-                                    val_fold_indices,
                                     feedback: str,
+                                    val_fold_indices = None,
                                     validation: Iterator[TTSplit] = None,
                                     random_state=42) -> ConfigurationSpace:
     """