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)