Merge 49253a0 into 4bdf714

bensadeghi · Aug 7, 2020 · b60338b · b60338b
2 parents 4bdf714 + 49253a0
commit b60338b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/classification/tree.jl b/src/classification/tree.jl
@@ -43,7 +43,7 @@ module treeclassifier
     # find an optimal split that satisfy the given constraints
     # (max_depth, min_samples_split, min_purity_increase)
     function _split!(
-            X                   :: Matrix{S},   # the feature array
+            X                   :: AbstractMatrix{S},   # the feature array
             Y                   :: AbstractVector{Int}, # the label array
             W                   :: AbstractVector{U},   # the weight vector
             purity_function     :: Function,
@@ -226,7 +226,7 @@ module treeclassifier
     end
 
     function check_input(
-            X                   :: Matrix{S},
+            X                   :: AbstractMatrix{S},
             Y                   :: AbstractVector{T},
             W                   :: AbstractVector{U},
             max_features        :: Int,

diff --git a/src/regression/tree.jl b/src/regression/tree.jl
@@ -42,7 +42,7 @@ module treeregressor
     # find an optimal split that satisfy the given constraints
     # (max_depth, min_samples_split, min_purity_increase)
     function _split!(
-            X                   :: Matrix{S}, # the feature array
+            X                   :: AbstractMatrix{S}, # the feature array
             Y                   :: AbstractVector{Float64}, # the label array
             W                   :: AbstractVector{U},
             node                :: NodeMeta{S}, # the node to split
@@ -229,7 +229,7 @@ module treeregressor
     end
 
     function check_input(
-            X                   :: Matrix{S},
+            X                   :: AbstractMatrix{S},
             Y                   :: AbstractVector{T},
             W                   :: AbstractVector{U},
             max_features        :: Int,