allow extrapolation in predictions

andrewfowlie · Jun 7, 2023 · 2a5802b · 2a5802b
1 parent 070b399
commit 2a5802b
Showing 1 changed file with 10 additions and 2 deletions.
diff --git 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):