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")