Source code for lmflow.pipeline.base_tuner

#!/usr/bin/env python
# coding=utf-8
""" BaseTuner: a subclass of BasePipeline.
"""

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
[docs] def tune(self, model, dataset): raise NotImplementedError(".tune is not implemented")