Source code for swarmauri.core.llms.IFit

from abc import ABC, abstractmethod



class IFit(ABC):
    """
    Interface for training models.
    """



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