Source code for esmvaltool.diag_scripts.mlr.models.krr

"""Kernel Ridge Regression model.

Use ``mlr_model_type: krr`` to use this MLR model in the recipe.

"""

import logging
import os

from sklearn.kernel_ridge import KernelRidge

from esmvaltool.diag_scripts.mlr.models import MLRModel

logger = logging.getLogger(os.path.basename(__file__))




@MLRModel.register_mlr_model('krr')
class KRRModel(MLRModel):
    """Kernel Ridge Regression model."""

    _CLF_TYPE = KernelRidge