Browse Source

[FIX] delete must test

tags/v0.3.2
liuht 2 years ago
parent
commit
ed0c4bd7fd
2 changed files with 1 additions and 3 deletions
  1. +1
    -1
      examples/dataset_table_workflow/base.py
  2. +0
    -2
      examples/dataset_table_workflow/hetero.py

+ 1
- 1
examples/dataset_table_workflow/base.py View File

@@ -111,7 +111,7 @@ class TableWorkflow:
save_path = os.path.join(save_root_path, f"{method_name}.json")
if method_name_full == "hetero_single_aug":
if test_info["force"] or recorder.should_test_method(user, idx, save_path):
if recorder.should_test_method(user, idx, save_path):
# * multi-process
queue = Queue()
processes = []


+ 0
- 2
examples/dataset_table_workflow/hetero.py View File

@@ -117,7 +117,6 @@ class HeterogeneousDatasetWorkflow(TableWorkflow):
def labeled_hetero_table_example(self):
logger.info("Total Items: %d" % len(self.market))
methods = ["user_model", "hetero_single_aug", "hetero_multiple_avg", "hetero_ensemble_pruning"]
methods_to_test = []
recorders = {method: Recorder() for method in methods + ["select_score", "oracle_score", "mean_score"]}

user = self.benchmark.name
@@ -170,7 +169,6 @@ class HeterogeneousDatasetWorkflow(TableWorkflow):
for method_name in methods:
logger.info(f"Testing method {method_name}")
test_info["method_name"] = method_name
test_info["force"] = method_name in methods_to_test
test_info.update(method_configs[method_name])
self.test_method(test_info, recorders, loss_func=loss_func_rmse)


Loading…
Cancel
Save