File tree Expand file tree Collapse file tree 2 files changed +5
-4
lines changed Expand file tree Collapse file tree 2 files changed +5
-4
lines changed Original file line number Diff line number Diff line change @@ -552,17 +552,18 @@ Mean Squared Error:
552
552
553
553
H(Q_m) = \frac {1 }{n_m} \sum _{y \in Q_m} (y - \bar {y}_m)^2
554
554
555
- Half Poisson deviance:
555
+ Mean Poisson deviance:
556
556
557
557
.. math ::
558
558
559
- H(Q_m) = \frac {1 }{n_m} \sum _{y \in Q_m} (y \log \frac {y}{\bar {y}_m}
559
+ H(Q_m) = \frac {2 }{n_m} \sum _{y \in Q_m} (y \log \frac {y}{\bar {y}_m}
560
560
- y + \bar {y}_m)
561
561
562
562
Setting `criterion="poisson" ` might be a good choice if your target is a count
563
563
or a frequency (count per some unit). In any case, :math: `y >= 0 ` is a
564
564
necessary condition to use this criterion. Note that it fits much slower than
565
- the MSE criterion.
565
+ the MSE criterion. For performance reasons the actual implementation minimizes
566
+ the half mean poisson deviance, i.e. the mean poisson deviance divided by 2.
566
567
567
568
Mean Absolute Error:
568
569
Original file line number Diff line number Diff line change @@ -1098,7 +1098,7 @@ class DecisionTreeRegressor(RegressorMixin, BaseDecisionTree):
1098
1098
mean squared error with Friedman's improvement score for potential
1099
1099
splits, "absolute_error" for the mean absolute error, which minimizes
1100
1100
the L1 loss using the median of each terminal node, and "poisson" which
1101
- uses reduction in Poisson deviance to find splits.
1101
+ uses reduction in the half mean Poisson deviance to find splits.
1102
1102
1103
1103
.. versionadded:: 0.18
1104
1104
Mean Absolute Error (MAE) criterion.
You can’t perform that action at this time.
0 commit comments