Class swarmauri_core.llms.IFit.IFit

swarmauri_core.llms.IFit.IFit

Bases: ABC

Interface for training models.

fit abstractmethod

fit(X_train, y_train, epochs, batch_size)

Train the model on the provided dataset.

Source code in swarmauri_core/llms/IFit.py

@abstractmethod
def fit(self, X_train, y_train, epochs: int, batch_size: int) -> None:
    """
    Train the model on the provided dataset.
    """
    pass