linearrf 1.2.5__tar.gz → 1.2.7__tar.gz

{linearrf-1.2.5/src/linearrf.egg-info → linearrf-1.2.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: linearrf
-Version: 1.2.5
+Version: 1.2.7
 Summary: A python libary to build Random Forests with Linear Models at the leaves.
 Author-email: Marian Biermann <marianbiermann@gmx.de>
 Project-URL: homepage, https://github.com/marianbiermann/lrf

{linearrf-1.2.5 → linearrf-1.2.7}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "linearrf"
-version = "1.2.5"
+version = "1.2.7"
 description = "A python libary to build Random Forests with Linear Models at the leaves."
 readme = "README.md"
 authors = [{ name = "Marian Biermann", email = "marianbiermann@gmx.de" }]

{linearrf-1.2.5 → linearrf-1.2.7/src/linearrf.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: linearrf
-Version: 1.2.5
+Version: 1.2.7
 Summary: A python libary to build Random Forests with Linear Models at the leaves.
 Author-email: Marian Biermann <marianbiermann@gmx.de>
 Project-URL: homepage, https://github.com/marianbiermann/lrf

{linearrf-1.2.5 → linearrf-1.2.7}/src/lrf/_base_lrf.py

@@ -113,7 +113,8 @@ class _LinearRandomForest:
         self.min_samples_leaf = self.leaf_samples_to_features_ratio * x.shape[1]
 
         # add intercept here and not inside linear model for performance reasons
-        x = np.insert(x, 0, 1, axis=1)
+        if self.linear_model is None:
+            x = np.insert(x, 0, 1, axis=1)
 
         # mtry is the same at every node — cache once instead of recomputing per split
         self._mtry = int(round(np.sqrt(x.shape[1] - 1)))
@@ -163,7 +164,8 @@ class _LinearRandomForest:
             if self.linear_model is None:
                 np.add(imp, np.abs(node.model.coef_[1:]), out=imp)
             else:
-                np.add(imp, np.abs(node.model.coef_), out=imp)
+                np.add(imp, np.abs(np.flatten(node.model.coef_)), out=imp)
+
         return imp
 
     @staticmethod