X-Git-Url: https://git.auder.net/?p=talweg.git;a=blobdiff_plain;f=pkg%2FR%2FF_Average.R;h=6cd2d6e93b3160dbcde46060180314b902a06f90;hp=d0b40b4ba69bd50b8c3f6c65ce49b7e5a432390b;hb=d2ab47a744d8fb29c03a76a7ca2368dae53f9a57;hpb=4d376294a6286ca1548d978055731dac175ffa3a

diff --git a/pkg/R/F_Average.R b/pkg/R/F_Average.R
index d0b40b4..6cd2d6e 100644
--- a/pkg/R/F_Average.R
+++ b/pkg/R/F_Average.R
@@ -17,25 +17,23 @@ AverageForecaster = R6::R6Class("AverageForecaster",
 	inherit = Forecaster,
 
 	public = list(
-		predictShape = function(data, today, memory, horizon, ...)
+		predictShape = function(data, today, memory, predict_from, horizon, ...)
 		{
-			avg = rep(0., horizon)
+			avg = rep(0., (horizon-predict_from+1))
 			first_day = max(1, today-memory)
-			index = today-7 + 1
+			index <- today
 			nb_no_na_series = 0
 			repeat
 			{
-				{
-					serie_on_horizon = data$getCenteredSerie(index)[1:horizon]
-					index = index - 7
-				};
+				index = index - 7
+				if (index < first_day)
+					break
+				serie_on_horizon = data$getCenteredSerie(index)[predict_from:horizon]
 				if (!any(is.na(serie_on_horizon)))
 				{
 					avg = avg + serie_on_horizon
 					nb_no_na_series = nb_no_na_series + 1
-				};
-				if (index < first_day)
-					break
+				}
 			}
 			avg / nb_no_na_series
 		}