logml.models.registry.neighbors

Classes

BaseNeighborsModel([params, logger])	Base class for sklearn.neighbors* models.
KNeighborsClassifierModel([params, logger])	Wrapper for sklearn.neighbors.KNeighborsClassifier.
KNeighborsRegressorModel([params, logger])	Wrapper for sklearn.neighbors.KNeighborsRegressor.

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

Bases: logml.models.base.BaseModel

Base class for sklearn.neighbors* models.

TAGS = ['neighbors']

FE_MODEL_ATTRIBUTE = None

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

Bases: logml.models.registry.neighbors.BaseNeighborsModel

Wrapper for sklearn.neighbors.KNeighborsRegressor.

TASK = 'regression'

F_MODEL

alias of sklearn.neighbors._regression.KNeighborsRegressor

DEFAULT_PARAMS = {'n_jobs': -1}

PARAMS_SPACE = {'leaf_size': [1, 5, 10], 'n_jobs': [2], 'n_neighbors': [1, 3, 5, 10], 'p': [2, 3], 'weights': ['uniform', 'distance']}

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

Bases: logml.models.registry.neighbors.BaseNeighborsModel

Wrapper for sklearn.neighbors.KNeighborsClassifier.

TASK = 'classification'

F_MODEL

alias of sklearn.neighbors._classification.KNeighborsClassifier

DEFAULT_PARAMS = {'n_jobs': -1}

PARAMS_SPACE = {'leaf_size': [1, 5, 10], 'n_jobs': [2], 'n_neighbors': [1, 3, 5, 10], 'p': [2, 3], 'weights': ['uniform', 'distance']}