Source code for niftynet.evaluation.classification_evaluator

# -*- coding: utf-8 -*-
"""
This module defines the specialized Evaluator for classification applications
All logic except default metrics is delegated to the parent class
"""

from __future__ import absolute_import, division, print_function

from niftynet.evaluation.base_evaluator import BaseEvaluator

class ClassificationEvaluator(BaseEvaluator):
    """
    Evaluator for ClassificationApplication
    """
    def default_evaluation_list(self):
        """
        :return:  list of metric names to compute by default
        """
        return ['classification_accuracy', 'roc_auc', 'roc']