"git@developer.sourcefind.cn:tianlh/lightgbm-dcu.git" did not exist on "0701a32da9fae7b2de8a01d702fa7d6abf36e836"
Commit 35d87188 authored by Guolin Ke's avatar Guolin Ke
Browse files

fix probability calculation (#28)

parent 7aaba32e
...@@ -364,7 +364,7 @@ double GBDT::Predict(const double* value) const { ...@@ -364,7 +364,7 @@ double GBDT::Predict(const double* value) const {
} }
// if need sigmoid transform // if need sigmoid transform
if (sigmoid_ > 0) { if (sigmoid_ > 0) {
ret = 1.0 / (1.0 + std::exp(-sigmoid_ * ret)); ret = 1.0 / (1.0 + std::exp(- 2.0f * sigmoid_ * ret));
} }
return ret; return ret;
} }
......
...@@ -57,7 +57,7 @@ public: ...@@ -57,7 +57,7 @@ public:
#pragma omp parallel for schedule(static) reduction(+:sum_loss) #pragma omp parallel for schedule(static) reduction(+:sum_loss)
for (data_size_t i = 0; i < num_data_; ++i) { for (data_size_t i = 0; i < num_data_; ++i) {
// sigmoid transform // sigmoid transform
score_t prob = 1.0f / (1.0f + std::exp(-sigmoid_ * score[i])); score_t prob = 1.0f / (1.0f + std::exp(-2.0f * sigmoid_ * score[i]));
// add loss // add loss
sum_loss += PointWiseLossCalculator::LossOnPoint(label_[i], prob); sum_loss += PointWiseLossCalculator::LossOnPoint(label_[i], prob);
} }
...@@ -65,7 +65,7 @@ public: ...@@ -65,7 +65,7 @@ public:
#pragma omp parallel for schedule(static) reduction(+:sum_loss) #pragma omp parallel for schedule(static) reduction(+:sum_loss)
for (data_size_t i = 0; i < num_data_; ++i) { for (data_size_t i = 0; i < num_data_; ++i) {
// sigmoid transform // sigmoid transform
score_t prob = 1.0f / (1.0f + std::exp(-sigmoid_ * score[i])); score_t prob = 1.0f / (1.0f + std::exp(-2.0f * sigmoid_ * score[i]));
// add loss // add loss
sum_loss += PointWiseLossCalculator::LossOnPoint(label_[i], prob) * weights_[i]; sum_loss += PointWiseLossCalculator::LossOnPoint(label_[i], prob) * weights_[i];
} }
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment