Fix shape of pred in DecisionTreeRegressor; closes issue #32

andosa / treeinterpreter

BSD 3-Clause "New" or "Revised" License

745 stars 140 forks source link

Fix shape of pred in DecisionTreeRegressor; closes issue #32 #39

Closed juliangilbey closed 3 years ago

juliangilbey commented 3 years ago

This closes #32 by ensuring that the shape of the output predictions from a DecisionTreeRegressor is a one-dimensional array rather than a 2D (n,1) array.