Class swarmauri_core.embeddings.ISaveModel.ISaveModel

swarmauri_core.embeddings.ISaveModel.ISaveModel

Bases: ABC

Interface to abstract the ability to save and load models.

save_model abstractmethod

save_model(path)

Saves the model to the specified directory.

Parameters: - path (str): The directory path where the model will be saved.

Source code in swarmauri_core/embeddings/ISaveModel.py

@abstractmethod
def save_model(self, path: str) -> None:
    """
    Saves the model to the specified directory.

    Parameters:
    - path (str): The directory path where the model will be saved.
    """
    pass

load_model abstractmethod

load_model(path)

Loads a model from the specified directory.

Parameters: - path (str): The directory path from where the model will be loaded.

Returns: - Returns an instance of the loaded model.

Source code in swarmauri_core/embeddings/ISaveModel.py

@abstractmethod
def load_model(self, path: str) -> Any:
    """
    Loads a model from the specified directory.

    Parameters:
    - path (str): The directory path from where the model will be loaded.

    Returns:
    - Returns an instance of the loaded model.
    """
    pass