diff --git a/benchmarks/ResidualAnalysis.py b/benchmarks/ResidualAnalysis.py index ebba1bd..db1fa1f 100644 --- a/benchmarks/ResidualAnalysis.py +++ b/benchmarks/ResidualAnalysis.py @@ -5,14 +5,14 @@ import numpy as np import pandas as pd import matplotlib as plt import matplotlib.pyplot as plt -from pyFTS.common import Transformations +from pyFTS.common import Transformations,Util def residuals(targets, forecasts, order=1): return np.array(targets[order:]) - np.array(forecasts[:-order]) -def plotResiduals(targets, forecasts, order=1, tam=[8, 8]): +def plotResiduals(targets, forecasts, order=1, tam=[8, 8], save=False, file=None): res = residuals(targets,forecasts,order) fig = plt.figure(figsize=tam) ax1 = fig.add_axes([0, 1, 0.9, 0.3]) # left, bottom, width, height @@ -22,3 +22,5 @@ def plotResiduals(targets, forecasts, order=1, tam=[8, 8]): ax3 = fig.add_axes([0, 0.3, 0.9, 0.3]) ax3.hist(res) + Util.showAndSaveImage(fig, file, save) +