Browse Source

[FIX] Fix "RKME" name

tags/v0.3.2
liuht 3 years ago
parent
commit
e34f9ee07c
2 changed files with 2 additions and 2 deletions
  1. +1
    -1
      learnware/specification/rkme.py
  2. +1
    -1
      learnware/specification/utils.py

+ 1
- 1
learnware/specification/rkme.py View File

@@ -31,7 +31,7 @@ logger = get_module_logger("rkme")


class RKMEStatSpecification(BaseStatSpecification):
"""Reduced-set Kernel Mean Embedding (RKME) Specification"""
"""Reduced Kernel Mean Embedding (RKME) Specification"""

def __init__(self, gamma: float = 0.1, cuda_idx: int = -1):
"""Initializing RKME parameters.


+ 1
- 1
learnware/specification/utils.py View File

@@ -16,7 +16,7 @@ def generate_rkme_spec(
cuda_idx: int = -1,
) -> RKMEStatSpecification:
"""
Interface for users to generate Reduced-set Kernel Mean Embedding (RKME) specification.
Interface for users to generate Reduced Kernel Mean Embedding (RKME) specification.
Return a RKMEStatSpecification object, use .save() method to save as json file.




Loading…
Cancel
Save