From f916e467570c3c2222b1b3ff2eb1944bf55be092 Mon Sep 17 00:00:00 2001 From: Gene Date: Tue, 26 Dec 2023 10:20:58 +0800 Subject: [PATCH] [MNT] merge label and unlabel tests --- examples/dataset_image_workflow/workflow.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/examples/dataset_image_workflow/workflow.py b/examples/dataset_image_workflow/workflow.py index b039446..736bc35 100644 --- a/examples/dataset_image_workflow/workflow.py +++ b/examples/dataset_image_workflow/workflow.py @@ -201,7 +201,9 @@ class ImageDatasetWorkflow: single_score_mat.append([best_acc] * repeated) user_model_score_mat.append(user_model_score_list) pruning_score_mat.append(reuse_pruning_score_list) - print(n_label, np.mean(user_model_score_mat[-1]), np.mean(pruning_score_mat[-1])) + print( + f"user_label_num: {n_label}, user_acc: {np.mean(user_model_score_mat[-1])}, pruning_acc: {np.mean(pruning_score_mat[-1])}" + ) logger.info(f"Saving Curves for User_{i}") user_curves_data = (single_score_mat, user_model_score_mat, pruning_score_mat)