X-Git-Url: https://git.auder.net/?p=talweg.git;a=blobdiff_plain;f=pkg%2FR%2FF_Neighbors.R;fp=pkg%2FR%2FF_Neighbors.R;h=ea27d5c12a002bdbc60ac20bba861c4fc028bb2c;hp=0295cd5c9420565ff01468d5d781dd93178700c5;hb=a3344f7591f6f4b3d337a69e4a568e9b16e33415;hpb=0730374e2981d7b950dced9f667713865b44b390

diff --git a/pkg/R/F_Neighbors.R b/pkg/R/F_Neighbors.R
index 0295cd5..ea27d5c 100644
--- a/pkg/R/F_Neighbors.R
+++ b/pkg/R/F_Neighbors.R
@@ -137,7 +137,7 @@ NeighborsForecaster = R6::R6Class("NeighborsForecaster",
 					days_in=tdays_cut, operational=opera)
 				nb_neighbs <- round( window[length(window)] )
 				# TODO: 10 == magic number
-				tdays <- .getConstrainedNeighbs(today, data, tdays, min_neighbs=nb_neighbs)
+				tdays <- .getConstrainedNeighbs(today, data, tdays, nb_neighbs, opera)
 				if (length(tdays) == 1)
 				{
 					if (final_call)
@@ -171,7 +171,7 @@ NeighborsForecaster = R6::R6Class("NeighborsForecaster",
 
 			if (simtype == "exo" || simtype == "mix")
 			{
-				distances2 <- .computeDistsExo(data, today, tdays)
+				distances2 <- .computeDistsExo(data, today, tdays, opera)
 
 				# Compute exogen similarities using the given window
 				window_exo = ifelse(simtype=="mix", window[2], window[1])
@@ -218,9 +218,9 @@ NeighborsForecaster = R6::R6Class("NeighborsForecaster",
 # @param min_neighbs Minimum number of points in a neighborhood
 # @param max_neighbs Maximum number of points in a neighborhood
 #
-.getConstrainedNeighbs = function(today, data, tdays, min_neighbs=10)
+.getConstrainedNeighbs = function(today, data, tdays, min_neighbs, opera)
 {
-	levelToday = data$getLevelHat(today)
+	levelToday = ifelse(opera, tail(data$getLevelHat(today),1), data$getLevel(today))
 	distances = sapply( tdays, function(i) abs(data$getLevel(i) - levelToday) )
 	#TODO: 1, +1, +3 : magic numbers
 	dist_thresh = 1
@@ -265,10 +265,13 @@ NeighborsForecaster = R6::R6Class("NeighborsForecaster",
 	})
 }
 
-.computeDistsExo <- function(data, today, tdays)
+.computeDistsExo <- function(data, today, tdays, opera)
 {
 	M = matrix( ncol=1+length(tdays), nrow=1+length(data$getExo(1)) )
-	M[,1] = c( data$getLevelHat(today), as.double(data$getExoHat(today)) )
+	if (opera)
+		M[,1] = c( tail(data$getLevelHat(today),1), as.double(data$getExoHat(today)) )
+	else
+		M[,1] = c( data$getLevel(today), as.double(data$getExo(today)) )
 	for (i in seq_along(tdays))
 		M[,i+1] = c( data$getLevel(tdays[i]), as.double(data$getExo(tdays[i])) )