From 68da04dc6bebd2483a79672eee6d081b5b942896 Mon Sep 17 00:00:00 2001 From: Peng Tan Date: Wed, 8 Nov 2023 21:15:19 +0800 Subject: [PATCH] [MNT] modify results print --- tests/test_market/test_hetero_market/test_hetero.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/tests/test_market/test_hetero_market/test_hetero.py b/tests/test_market/test_hetero_market/test_hetero.py index 19c5ad5..2d367ac 100644 --- a/tests/test_market/test_hetero_market/test_hetero.py +++ b/tests/test_market/test_hetero_market/test_hetero.py @@ -309,8 +309,11 @@ class TestMarket(unittest.TestCase): mixture_learnware_list, ) = hetero_market.search_learnware(user_info) + # print search results + for score, learnware in zip(sorted_score_list, single_learnware_list): + print(f"score: {score}, learnware_id: {learnware.id}") + # model reuse - print([learnware.id for learnware in single_learnware_list]) reuser=HeteroMapTableReuser(single_learnware_list[0], task_type='regression') reuser.fit(user_spec) y_pred=reuser.predict(X)