diff --git a/rep/report/_base.py b/rep/report/_base.py index 3d9e985f..d3439bae 100644 --- a/rep/report/_base.py +++ b/rep/report/_base.py @@ -187,8 +187,11 @@ def compute_metric(self, metric, mask=None): """ Compute metric value - :param metric: class with __call__(self, y_true, prob, sample_weight=None) or function - :param mask: mask which points we should use + :param metric: function like object with:: + + __call__(self, y_true, prob, sample_weight=None) + + :param mask: mask, points we should use :type mask: None or array-like or str or function(pandas.DataFrame) :return: metric value for each estimator diff --git a/rep/report/classification.py b/rep/report/classification.py index 24c3bc6f..0026a76f 100644 --- a/rep/report/classification.py +++ b/rep/report/classification.py @@ -42,7 +42,8 @@ class ClassificationReport(AbstractReport): Parameters: ----------- - :param dict[str, Classifier] classifiers: estimators + :param classifiers: estimators + :type classifiers: dict[str, Classifier] :param LabeledDataStorage lds: data """ def __init__(self, classifiers, lds):