Source code for lightmlboard.metrics

"""
Implements metrics.


:githublink:`%|py|5`
"""
import sklearn.metrics as skmetrics
from .classification import roc_auc_score_micro, roc_auc_score_macro, reshape, multi_label_jaccard
from .regression import mse
from .regression_custom import l1_reg_max


def sklearn_metric(met, exp, val):
    """
    Looks into metrics available in
    :epkg:`scikit-learn:metrics`.

    :param      met:     function name
    :param      exp:     expected values
    :param      val:     values
    :return:             number


    :githublink:`%|py|20`
    """
    if isinstance(val, str):
        raise TypeError("val must be a container of floats")
    if isinstance(exp, str):
        raise TypeError("exp must be a container of floats")
    if hasattr(skmetrics, met):
        f = getattr(skmetrics, met)
        exp, val = reshape(exp, val)
        return f(exp, val)
    else:
        raise AttributeError("Unable to find metric '{0}'.".format(met))