Browse Source

Merge branch 'table_benchmark_update' of https://github.com/Learnware-LAMDA/Learnware into table_benchmark_update

tags/v0.3.2
liuht 2 years ago
parent
commit
dd93185dc6
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      examples/dataset_table_workflow/homo.py

+ 2
- 2
examples/dataset_table_workflow/homo.py View File

@@ -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"))


Loading…
Cancel
Save