File size: 522 Bytes
3455d37
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
#!/usr/bin/env python
# coding=utf-8
""" BaseTuner: a subclass of BasePipeline.
"""

from lmflow.pipeline.base_pipeline import BasePipeline


class BaseTuner(BasePipeline):
    """ A subclass of BasePipeline which is tunable.
    """
    def __init__(self, *args, **kwargs):
        pass

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

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