Browse Source

[MNT] Fix typo, modify generate spec interface

tags/v0.3.2
bxdd 3 years ago
parent
commit
ee25c45945
3 changed files with 9 additions and 4 deletions
  1. +2
    -2
      examples/example_market_db/example_db.py
  2. +1
    -1
      learnware/market/easy.py
  3. +6
    -1
      learnware/specification/base.py

+ 2
- 2
examples/example_market_db/example_db.py View File

@@ -117,7 +117,7 @@ def test_market():
print("Available ids:", curr_inds)


def test_search_sementics():
def test_search_semantics():
easy_market = EasyMarket()
print("Total Item:", len(easy_market))

@@ -175,4 +175,4 @@ if __name__ == "__main__":
prepare_learnware(learnware_num)
test_market()
test_stat_search()
test_search_sementics()
test_search_semantics()

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

@@ -244,7 +244,7 @@ class EasyMarket(BaseMarket):
learnware_num = len(learnware_list)
if learnware_num == 0:
return [], []
if learnware_num < search_num:
if learnware_num < search_num:
logger.warning("Available Learnware num less than search_num")
search_num = learnware_num



+ 6
- 1
learnware/specification/base.py View File

@@ -6,7 +6,12 @@ class BaseStatSpecification:
def __init__(self):
pass

def generate_stat_spec_from_data(self, X: np.ndarray):
def generate_stat_spec_from_data(self, **kwargs):
"""Construct reduced set from raw dataset using iterative optimization

- kwargs may include the feature, label and model
- kwargs also can include hyperparameter for specifaction generation
"""
raise NotImplementedError("generate_stat_spec_from_data is not implemented")

def save(self, filepath: str):


Loading…
Cancel
Save