From 9d3b423900d520a18450aeb5f82ad2f931d0c19e Mon Sep 17 00:00:00 2001 From: bxdd Date: Tue, 18 Apr 2023 21:46:53 +0800 Subject: [PATCH] [MNT] Modify base reuser --- learnware/learnware/base.py | 1 + learnware/learnware/reuse.py | 13 +++++++++++-- 2 files changed, 12 insertions(+), 2 deletions(-) diff --git a/learnware/learnware/base.py b/learnware/learnware/base.py index 3b30b8a..4ad4991 100644 --- a/learnware/learnware/base.py +++ b/learnware/learnware/base.py @@ -61,3 +61,4 @@ class BaseReuser: np.ndarray The final prediction for user data with reused learnware """ + raise NotImplementedError("The predict method is not implemented!") diff --git a/learnware/learnware/reuse.py b/learnware/learnware/reuse.py index 85e6a97..19a06f0 100644 --- a/learnware/learnware/reuse.py +++ b/learnware/learnware/reuse.py @@ -10,13 +10,22 @@ from .base import BaseReuser from ..specification import RKMEStatSpecification from ..logger import get_module_logger -logger = get_module_logger("BaseReuser") +logger = get_module_logger("Reuser") class JobSelectorReuser(BaseReuser): - """Baseline Multiple Learnware Reuse uing Job Selector Method""" + """Baseline Multiple Learnware Reuser uing Job Selector Method""" def __init__(self, learnware_list: List[Learnware], herding_num: int = 1000): + """The initialization method for job selector reuser + + Parameters + ---------- + learnware_list : List[Learnware] + The learnware list, which should have RKME Specification for each learnweare + herding_num : int, optional + The , by default 1000 + """ super(JobSelectorReuser, self).__init__(learnware_list) self.herding_num = herding_num