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