add the regression benchmark

This commit is contained in:
Zhenwen Dai 2015-08-27 17:04:52 +01:00
parent f221a3b1fa
commit 60aa865631
7 changed files with 310 additions and 2 deletions

View file

@ -0,0 +1,21 @@
# 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())