GPy/benchmarks/regression/evaluation.py

# Copyright (c) 2015, Zhenwen Dai
# Licensed under the BSD 3-clause license (see LICENSE.txt)

import abc
import numpy as np


class Evaluation(object):
    __metaclass__ = abc.ABCMeta

    @abc.abstractmethod
    def evaluate(self, gt, pred):
        """Compute a scalar for access the performance"""
        return None


class RMSE(Evaluation):
    "Rooted Mean Square Error"
    name = "RMSE"

    def evaluate(self, gt, pred):
        return np.sqrt(np.square(gt - pred).astype(float).mean())