Skip to content

Commit ce65d71

Browse files
author
Max
committed
Fixed an issue where the recommendations in explicit_evaler.py were not converted to DataFrames first
1 parent 6f0127d commit ce65d71

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

lkauto/explicit/explicit_evaler.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -151,7 +151,7 @@ def evaluate(self, config_space: ConfigurationSpace) -> Tuple[float, Pipeline]:
151151
best_model = fit_pipeline
152152

153153
error_metric = np.append(error_metric, error_results)
154-
validation_data = pd.concat([validation_data, recs], ignore_index=True)
154+
validation_data = pd.concat([validation_data, recs.to_df()], ignore_index=True)
155155
else:
156156
for fold in range(self.split_folds):
157157
validation_train = self.train
@@ -184,7 +184,7 @@ def evaluate(self, config_space: ConfigurationSpace) -> Tuple[float, Pipeline]:
184184
best_model = fit_pipeline
185185

186186
error_metric = np.append(error_metric, error_results)
187-
validation_data = pd.concat([validation_data, recs], ignore_index=True)
187+
validation_data = pd.concat([validation_data, recs.to_df()], ignore_index=True)
188188

189189
# Save validation data for reproducibility and ensembling
190190
self.top_n_runs = update_top_n_runs(config_space=config_space,

0 commit comments

Comments
 (0)