From 368229e9792f2a488fec4c1e898259e0c9926061 Mon Sep 17 00:00:00 2001 From: shihy Date: Mon, 20 Nov 2023 21:54:12 +0800 Subject: [PATCH 1/2] [MNT] Improve Scoring Function for Image Learnware --- learnware/market/easy/searcher.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/learnware/market/easy/searcher.py b/learnware/market/easy/searcher.py index 0748614..3e7720e 100644 --- a/learnware/market/easy/searcher.py +++ b/learnware/market/easy/searcher.py @@ -586,6 +586,12 @@ class EasyStatSearcher(BaseSearcher): weight_list = [] mixture_learnware_list = [] + # Special Transform for ImageSpecification + if self.stat_spec_type == "RKEMImageSpecification": + sorted_dist_list = [np.exp(-d / 0.00005) for d in sorted_dist_list] + if mixture_dist is not None: + mixture_dist = np.exp(-mixture_dist / 0.00005) + if mixture_dist is None: sorted_score_list = self._convert_dist_to_score(sorted_dist_list) mixture_score = None From e21ba51f3d34398a8def8f11a82259a65290f552 Mon Sep 17 00:00:00 2001 From: shihy Date: Mon, 20 Nov 2023 21:58:03 +0800 Subject: [PATCH 2/2] [MNT] Fix typo --- learnware/market/easy/searcher.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/learnware/market/easy/searcher.py b/learnware/market/easy/searcher.py index 3e7720e..15b7262 100644 --- a/learnware/market/easy/searcher.py +++ b/learnware/market/easy/searcher.py @@ -587,7 +587,7 @@ class EasyStatSearcher(BaseSearcher): mixture_learnware_list = [] # Special Transform for ImageSpecification - if self.stat_spec_type == "RKEMImageSpecification": + if self.stat_spec_type == "RKMEImageSpecification": sorted_dist_list = [np.exp(-d / 0.00005) for d in sorted_dist_list] if mixture_dist is not None: mixture_dist = np.exp(-mixture_dist / 0.00005)