Commit a92ca5a2 authored by Jérôme BUISINE's avatar Jérôme BUISINE
Browse files

computation of mae

parent 1035bf1a
......@@ -59,6 +59,9 @@ class FitterAnalysis(Analysis):
def coefficient_of_determination(self, surrogate):
return r2_score(self.ytest, surrogate.predict(self.xtest))
def mae(self, surrogate):
return np.mean(np.abs(self.ytest - surrogate.predict(self.xtest)))
#########################################
# OptimizerAnalysis
#########################################
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment