Skip to content

Commit 2a6ba95

Browse files
gradient inverse
1 parent 04a316f commit 2a6ba95

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
@@ -461,7 +461,8 @@ VectorXd APLRRegressor::differentiate_predictions()
461461
else if(link_function=="inverse")
462462
{
463463
VectorXd transformed_linear_predictor{transform_linear_predictor_to_negative(linear_predictor_current)};
464-
return 1.0 * transformed_linear_predictor.array().pow(-2);
464+
double scaling{std::pow(transformed_linear_predictor.mean(),2)};
465+
return scaling * transformed_linear_predictor.array().pow(-2);
465466
}
466467
return VectorXd(0);
467468
}

0 commit comments

Comments
 (0)