mirror of
https://github.com/SheffieldML/GPy.git
synced 2026-04-25 04:46:23 +02:00
21 lines
510 B
Python
21 lines
510 B
Python
# 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(np.float).mean())
|
|
|