Browse Source

[MNT] modify results print

tags/v0.3.2
Peng Tan 2 years ago
parent
commit
68da04dc6b
1 changed files with 4 additions and 1 deletions
  1. +4
    -1
      tests/test_market/test_hetero_market/test_hetero.py

+ 4
- 1
tests/test_market/test_hetero_market/test_hetero.py View File

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


Loading…
Cancel
Save