Skip to content

Commit 5c5250e

Browse files
authored
Merge pull request #4777 from cxysteven/develop
lambdaCost layer's bug fix
2 parents 44ed21e + 60c2536 commit 5c5250e

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

paddle/gserver/layers/CostLayer.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -462,8 +462,8 @@ void LambdaCost::calcGrad(const real* outputScore,
462462
real score_j = score[index_j];
463463
real dcgDif = 0;
464464
if (j < sortSize) {
465-
dcgDif = (std::pow(2, score_i) - std::pow(2, score_j)) /
466-
(std::log(i + 2) - std::log(j + 2));
465+
dcgDif = (std::pow(2, score_i) - std::pow(2, score_j)) *
466+
(1 / std::log(i + 2) - 1 / std::log(j + 2));
467467
} else {
468468
dcgDif =
469469
(std::pow(2, score_i) - std::pow(2, score_j)) / std::log(i + 2);

0 commit comments

Comments
 (0)