Browse Source

[FIX] Fix bugs

tags/v0.3.2
Gene 3 years ago
parent
commit
e64adb4837
2 changed files with 2 additions and 2 deletions
  1. +1
    -1
      examples/example_pfs/main.py
  2. +1
    -1
      learnware/market/easy.py

+ 1
- 1
examples/example_pfs/main.py View File

@@ -141,7 +141,7 @@ class PFSDatasetWorkflow:
single_learnware_list,
mixture_score,
mixture_learnware_list,
) = easy_market.search_learnware(user_info)
) = easy_market.search_learnware(user_info, search_method = "auto")

print(f"search result of user{idx}:")
print(


+ 1
- 1
learnware/market/easy.py View File

@@ -385,7 +385,7 @@ class EasyMarket(BaseMarket):
if len(mixture_list) <= 1:
mixture_list = [learnware_list[sort_by_weight_idx_list[0]]]
mixture_weight = [1]
mmd_dist = user_rkme.dist(mixture_list)
mmd_dist = user_rkme.dist(mixture_list[0])
else:
if len(mixture_list) > max_search_num:
mixture_list = mixture_list[:max_search_num]


Loading…
Cancel
Save