Skip to content

Commit e27f116

Browse files
tweedie scaling wip
1 parent 7e743ad commit e27f116

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

cpp/APLRRegressor.h

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -456,7 +456,8 @@ VectorXd APLRRegressor::differentiate_predictions()
456456
else if(link_function=="tweedie")
457457
{
458458
VectorXd transformed_linear_predictor{transform_linear_predictor_to_negative(linear_predictor_current)};
459-
return ((1-tweedie_power)*transformed_linear_predictor.array()).pow(tweedie_power/(1-tweedie_power));
459+
double scaling{std::pow((1-tweedie_power)*transformed_linear_predictor.mean(),tweedie_power/(1-tweedie_power))};
460+
return scaling*((1-tweedie_power)*transformed_linear_predictor.array()).pow(tweedie_power/(1-tweedie_power));
460461
}
461462
else if(link_function=="inverse")
462463
{

0 commit comments

Comments
 (0)