diff --git a/python/mxnet/metric.py b/python/mxnet/metric.py index a8a8230a9de1..af9224a11292 100644 --- a/python/mxnet/metric.py +++ b/python/mxnet/metric.py @@ -54,8 +54,10 @@ def update(self, label, pred): self.sum_metric += numpy.sum(pred_label == label) self.num_inst += label.size +# pylint: disable=pointless-string-statement +""" class LogLoss(EvalMetric): - """Calculate logloss""" + # remove because it because it is too slow def __init__(self): self.eps = 1e-15 super(LogLoss, self).__init__('logloss') @@ -70,6 +72,8 @@ def update(self, label, pred): p = max(min(p, 1 - self.eps), self.eps) self.sum_metric += -numpy.log(p) self.num_inst += label.size +""" +# pylint: enable=pointless-string-statement class CustomMetric(EvalMetric): """Custom evaluation metric that takes a NDArray function. @@ -128,7 +132,5 @@ def create(metric): raise TypeError('metric should either be callable or str') if metric == 'acc' or metric == 'accuracy': return Accuracy() - elif metric == 'logloss': - return LogLoss() else: raise ValueError('Cannot find metric %s' % metric)