Source code for optunaz.utils.enums.interface_enum

from optunaz.utils.enums.configuration_enum import ConfigurationEnum


class InterfaceEnum:
    """This "Enum" serves as storage for the interface specifications."""

    # use the fixed names of the configuration enum
    _CE = ConfigurationEnum()

    # set the sets for the algorithms
    SKLEARN_SET = {
        _CE.ALGORITHMS_LASSO,
        _CE.ALGORITHMS_RIDGE,
        _CE.ALGORITHMS_LOGISTICREGRESSION,
        _CE.ALGORITHMS_PLSREGRESSION,
        _CE.ALGORITHMS_RFCLASSIFIER,
        _CE.ALGORITHMS_RFREGRESSOR,
        _CE.ALGORITHMS_SVC,
        _CE.ALGORITHMS_SVR,
        _CE.ALGORITHMS_ADABOOSTCLASSIFIER,
    }
    XGBOOST_SET = {_CE.ALGORITHMS_XGBREGRESSOR}
    CHEMPROP_SET = {
        _CE.ALGORITHMS_CHEMPROP,
        _CE.ALGORITHMS_CHEMPROP_REGRESSOR,
        _CE.ALGORITHMS_CHEMPROP_CLASSIFIER,
    }
    PRF_SET = {_CE.ALGORITHMS_PRF}
    CALIBRATED_SET = {
        _CE.ALGORITHMS_MAPIE,
        _CE.ALGORITHMS_CALIBRATEDCLASSIFIERCV,
    }

    # try to find the internal value and return
    def __getattr__(self, name):
        if name in self:
            return name
        raise AttributeError

    # prohibit any attempt to set any values
    def __setattr__(self, key, value):
        raise ValueError("No changes allowed.")