From 2a5802b9e54be26a4a0caa921a578ec983bf9102 Mon Sep 17 00:00:00 2001 From: Andrew Fowlie Date: Wed, 7 Jun 2023 09:32:38 +0800 Subject: [PATCH] allow extrapolation in predictions --- src/kingpin/model.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/src/kingpin/model.py b/src/kingpin/model.py index f8b168d..8dcb662 100644 --- a/src/kingpin/model.py +++ b/src/kingpin/model.py @@ -42,8 +42,16 @@ def where(self, arr, interval): """ :return: Partitions according to an interval """ - lower = interval[0] * self.x_data_delta + self.x_data_min - upper = interval[1] * self.x_data_delta + self.x_data_min + if interval[0] <= 0.: + lower = -np.inf + else: + lower = interval[0] * self.x_data_delta + self.x_data_min + + if interval[1] >= 1.: + upper = np.inf + else: + upper = interval[1] * self.x_data_delta + self.x_data_min + return (arr <= upper) & (arr >= lower) def min_data_points(self, intervals):