# Path: utility/runtype.py
# class RunType(Enum):
#     RUN_TRAIN_VALIDATION = 1
#     RUN_CROSS_VALIDATION = 2
#     RUN_GRID_SEARCH      = 3
#     RUN_TEST_PREDICTION = 4


from utility.runtype import RunType


class ModelFramework(object):
    """
    This is meant to be a base class that should be inherited by concrete models,and they should implement
    get_model_input
    run_train_validation
    run_croos_validation
    run_grid_search
    """
    def __init__(self):
        self.dtrain = None
        self.folds_params = None 

# Path: utility/runtype.py
# class RunType(Enum):
#     RUN_TRAIN_VALIDATION = 1
#     RUN_CROSS_VALIDATION = 2
#     RUN_GRID_SEARCH      = 3
#     RUN_TEST_PREDICTION = 4


# Path: utility/modelframework.py
from utility.runtype import RunType



class ModelFramework(object):
    """
    This is meant to be a base class that should be inherited by concrete models,and they should implement
    get_model_input
    run_train_validation
    run_croos_validation
    run_grid_search
    """
    def __init__(self):
        self.dtrain = None
        self.folds_params = None 

        self.run_type = RunType.RUN_GRID_SEARCH