diff --git a/examples/dataset_table_workflow/homo.py b/examples/dataset_table_workflow/homo.py index 1fd7403..0b44f8b 100644 --- a/examples/dataset_table_workflow/homo.py +++ b/examples/dataset_table_workflow/homo.py @@ -115,7 +115,7 @@ class HomogeneousDatasetWorkflow(TableWorkflow): for idx in range(self.benchmark.user_num): test_x, test_y = self.benchmark.get_test_data(user_ids=idx) test_x, test_y = test_x.values, test_y.values - + train_x, train_y = self.benchmark.get_train_data(user_ids=idx) train_x, train_y = train_x.values, train_y.values train_subsets = self.get_train_subsets(homo_n_labeled_list, homo_n_repeat_list, train_x, train_y) @@ -155,7 +155,7 @@ class HomogeneousDatasetWorkflow(TableWorkflow): test_info["method_name"] = method_name test_info.update(method_configs[method_name]) self.test_method(test_info, recorders, loss_func=loss_func_rmse) - + for method, recorder in recorders.items(): recorder.save(os.path.join(self.curves_result_path, f"{user}/{user}_{method}_performance.json"))