Stop splitting if there is no split that reduces loss

1 files changed, 3 insertions, 1 deletions

diff --git a/lib/parameters.py b/lib/parameters.py
index a32eb9f..ff0fdda 100644
--- a/lib/parameters.py
+++ b/lib/parameters.py
@@ -1306,7 +1306,9 @@ class ModelAttribute:
             assert not np.any(np.isnan(children))
             loss.append(np.sum(children))
 
-        if np.all(np.isinf(loss)):
+        if np.all(np.isinf(loss)) or np.min(loss) >= np.sum(
+            (np.array(data) - np.mean(data)) ** 2
+        ):
             if ffs_feasible:
                 # try generating a function. if it fails, model_function is a StaticFunction.
                 ma = ModelAttribute(