scikit-learn
diff --git a/‎sklearn/__init__.py
Lines changed: 1 addition & 1 deletion b/‎sklearn/__init__.py
Lines changed: 1 addition & 1 deletion
diff --git a/‎sklearn/tree/_criterion.pyx
Lines changed: 1 addition & 0 deletions b/‎sklearn/tree/_criterion.pyx
Lines changed: 1 addition & 0 deletions
diff --git a/‎sklearn/tree/_splitter.pxd
Lines changed: 6 additions & 1 deletion b/‎sklearn/tree/_splitter.pxd
Lines changed: 6 additions & 1 deletion
@@ -37,7 +37,7 @@
 # Dev branch marker is: 'X.Y.dev' or 'X.Y.devN' where N is an integer.
 # 'X.Y.dev0' is the canonical version of 'X.Y.dev'
 #
-__version__ = '0.19.dev0'
+__version__ = '0.19.dev1'
 
 
 try:
 
@@ -969,6 +969,7 @@ cdef class MSE(RegressionCriterion):
         impurity_left[0] /= self.n_outputs
         impurity_right[0] /= self.n_outputs
 
+
 cdef class MAE(RegressionCriterion):
     """Mean absolute error impurity criterion
 
 
@@ -19,6 +19,7 @@ ctypedef np.npy_float64 DOUBLE_t         # Type of y, sample_weight
 ctypedef np.npy_intp SIZE_t              # Type for indices and counters
 ctypedef np.npy_int32 INT32_t            # Signed 32 bit integer
 ctypedef np.npy_uint32 UINT32_t          # Unsigned 32 bit integer
+ctypedef np.npy_uint64 UINT64_t          # Unsigned 64 bit integer
 
 cdef struct SplitRecord:
     # Data to track sample split
@@ -30,6 +31,9 @@ cdef struct SplitRecord:
     double improvement     # Impurity improvement given parent node.
     double impurity_left   # Impurity of the left split.
     double impurity_right  # Impurity of the right split.
+    SIZE_t n_categories    # Num. of categories of the feature; -1 if not categorical.
+    UINT64_t split_map     # bitmap guiding how to split; 1 means right node.
+
 
 cdef class Splitter:
     # The splitter searches in the input space for a feature and a threshold
@@ -83,7 +87,8 @@ cdef class Splitter:
     # Methods
     cdef void init(self, object X, np.ndarray y,
                    DOUBLE_t* sample_weight,
-                   np.ndarray X_idx_sorted=*) except *
+                   np.ndarray X_idx_sorted=*,
+                   np.ndarray categorical_features=*) except *
 
     cdef void node_reset(self, SIZE_t start, SIZE_t end,
                          double* weighted_n_node_samples) nogil
Original file line number	Diff line number	Diff line change
`@@ -37,7 +37,7 @@`
`37`	`37`	`# Dev branch marker is: 'X.Y.dev' or 'X.Y.devN' where N is an integer.`
`38`	`38`	`# 'X.Y.dev0' is the canonical version of 'X.Y.dev'`
`39`	`39`	`#`
`40`		`-__version__ = '0.19.dev0'`
	`40`	`+__version__ = '0.19.dev1'`
`41`	`41`
`42`	`42`
`43`	`43`	`try:`