Fix gamma neg log likelihood.

dmlc · Sep 29, 2021 · ee64178 · ee64178
1 parent b2d8431
commit ee64178
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/metric/elementwise_metric.cu b/src/metric/elementwise_metric.cu
@@ -309,8 +309,7 @@ struct EvalGammaNLogLik {
     float constexpr kPsi = 1.0;
     bst_float theta = -1. / py;
     bst_float a = kPsi;
-    // b = -std::log(-theta);
-    float b = 1.0f;
+    float b = -std::log(-theta);
     // c = 1. / kPsi * std::log(y/kPsi) - std::log(y) - common::LogGamma(1. / kPsi);
     //   = 1.0f      * std::log(y)      - std::log(y) - 0 = 0
     float c = 0;