Source code for niftynet.evaluation.regression_evaluator

# -*- coding: utf-8 -*-
"""
This module defines the specialized Evaluator for segmentation 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 RegressionEvaluator(BaseEvaluator):
    """
    Evaluator for RegressionApplication
    """
    def default_evaluation_list(self):
        """
        :return:  list of metric names to compute by default
        """
        return ['mse', 'rmse', 'mae']