Source code for lmflow.pipeline.base_tuner

#!/usr/bin/env python
"""BaseTuner: a subclass of BasePipeline."""

from abc import abstractmethod

from lmflow.pipeline.base_pipeline import BasePipeline



[docs]
class BaseTuner(BasePipeline):
    """A subclass of BasePipeline which is tunable."""

    def __init__(self, *args, **kwargs):
        pass


[docs]
    def _check_if_tunable(self, model, dataset):
        # TODO: check if the model is tunable and dataset is compatible
        pass


    @abstractmethod

[docs]
    def tune(self, model, dataset):
        raise NotImplementedError(".tune is not implemented")