Update and fix docs

Signed-off-by: Adam Li <adam2392@gmail.com>
neurodata · Jul 20, 2023 · 28df968 · 28df968
1 parent 0b077cf
commit 28df968
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 11 deletions.
diff --git a/examples/plot_projection_matrices.py b/examples/plot_projection_matrices.py
@@ -61,6 +61,7 @@
 
 boundary = None
 feature_weight = None
+missing_value_feature_mask = None
 
 # initialize some dummy data
 X = np.repeat(np.arange(25).astype(np.float32), 5).reshape(5, -1)
@@ -86,8 +87,7 @@
     min_samples_leaf,
     min_weight_leaf,
     random_state,
-    None,
-    None,
+    missing_value_feature_mask,
     min_patch_dims,
     max_patch_dims,
     dim_contiguous,
@@ -132,8 +132,7 @@
     min_samples_leaf,
     min_weight_leaf,
     random_state,
-    None,
-    None,
+    missing_value_feature_mask,
     min_patch_dims,
     max_patch_dims,
     dim_contiguous,
@@ -182,8 +181,7 @@
     min_samples_leaf,
     min_weight_leaf,
     random_state,
-    None,
-    None,
+    missing_value_feature_mask,
     min_patch_dims,
     max_patch_dims,
     dim_contiguous,
@@ -246,8 +244,7 @@
     min_samples_leaf,
     min_weight_leaf,
     random_state,
-    None,
-    None,
+    missing_value_feature_mask,
     min_patch_dims,
     max_patch_dims,
     dim_contiguous,
@@ -285,8 +282,7 @@
     min_samples_leaf,
     min_weight_leaf,
     random_state,
-    None,
-    None,
+    missing_value_feature_mask,
     min_patch_dims,
     max_patch_dims,
     dim_contiguous,

diff --git a/sktree/tree/_classes.py b/sktree/tree/_classes.py
@@ -234,7 +234,7 @@ def _build_tree(
         max_depth,
         random_state,
     ):
-        monotonic_cst = None
+        # monotonic_cst = None
         # self.n_categories_ = np.array(
         #     [
         #         np.int32(X[:, i].max()) + 1 if i in categorical else -1