logml.models.registry.dummy

Classes

BaseDummyModel([params, logger])	Base class for dummy models without inner CV.
DummyClassifierModel([params, logger])	Wrapper for sklearn.dummy.DummyClassifier.
DummyRegressorModel([params, logger])	Wrapper for sklearn.dummy.DummyRegressor.
DummySurvivalEstimator(*[, strategy])	Analog of DummyRegressor model for survival problems.
DummySurvivalModel([params, logger])	Dummy survival model.

class logml.models.registry.dummy.BaseDummyModel(params: Optional[dict] = None, logger=None)

Bases: logml.models.base.BaseModel

Base class for dummy models without inner CV. TAGS are set.

TAGS = ['dummy']

FE_MODEL_ATTRIBUTE = None

class logml.models.registry.dummy.DummyRegressorModel(params: Optional[dict] = None, logger=None)

Bases: logml.models.registry.dummy.BaseDummyModel

Wrapper for sklearn.dummy.DummyRegressor.

TASK = 'regression'

F_MODEL

alias of sklearn.dummy.DummyRegressor

DEFAULT_PARAMS = {'strategy': 'mean'}

PARAMS_SPACE = {'quantile': [0.6], 'strategy': ['mean', 'median', 'quantile']}

class logml.models.registry.dummy.DummyClassifierModel(params: Optional[dict] = None, logger=None)

Bases: logml.models.registry.dummy.BaseDummyModel

Wrapper for sklearn.dummy.DummyClassifier.

TASK = 'classification'

F_MODEL

alias of sklearn.dummy.DummyClassifier

DEFAULT_PARAMS = {'random_state': 42, 'strategy': 'stratified'}

PARAMS_SPACE = {'strategy': ['stratified', 'most_frequent', 'prior', 'uniform']}

class logml.models.registry.dummy.DummySurvivalEstimator(*, strategy='median', **kwargs)

Bases: sklearn.dummy.DummyRegressor, sksurv.base.SurvivalAnalysisMixin

Analog of DummyRegressor model for survival problems.

fit(X, y, **kwargs)

Fit estimator.

predict(X, **kwargs)

Predict.

class logml.models.registry.dummy.DummySurvivalModel(params: Optional[dict] = None, logger=None)

Bases: logml.models.registry.dummy.BaseDummyModel, sksurv.base.SurvivalAnalysisMixin

Dummy survival model.

DEFAULT_PARAMS = {}

PARAMS_SPACE = None

TASK = 'survival'

F_MODEL

alias of logml.models.registry.dummy.DummySurvivalEstimator