Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
tianlh
LightGBM-DCU
Commits
35d87188
Commit
35d87188
authored
Oct 21, 2016
by
Guolin Ke
Browse files
fix probability calculation (#28)
parent
7aaba32e
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
3 additions
and
3 deletions
+3
-3
src/boosting/gbdt.cpp
src/boosting/gbdt.cpp
+1
-1
src/metric/binary_metric.hpp
src/metric/binary_metric.hpp
+2
-2
No files found.
src/boosting/gbdt.cpp
View file @
35d87188
...
...
@@ -364,7 +364,7 @@ double GBDT::Predict(const double* value) const {
}
// if need sigmoid transform
if
(
sigmoid_
>
0
)
{
ret
=
1.0
/
(
1.0
+
std
::
exp
(
-
sigmoid_
*
ret
));
ret
=
1.0
/
(
1.0
+
std
::
exp
(
-
2.0
f
*
sigmoid_
*
ret
));
}
return
ret
;
}
...
...
src/metric/binary_metric.hpp
View file @
35d87188
...
...
@@ -57,7 +57,7 @@ public:
#pragma omp parallel for schedule(static) reduction(+:sum_loss)
for
(
data_size_t
i
=
0
;
i
<
num_data_
;
++
i
)
{
// sigmoid transform
score_t
prob
=
1.0
f
/
(
1.0
f
+
std
::
exp
(
-
sigmoid_
*
score
[
i
]));
score_t
prob
=
1.0
f
/
(
1.0
f
+
std
::
exp
(
-
2.0
f
*
sigmoid_
*
score
[
i
]));
// add loss
sum_loss
+=
PointWiseLossCalculator
::
LossOnPoint
(
label_
[
i
],
prob
);
}
...
...
@@ -65,7 +65,7 @@ public:
#pragma omp parallel for schedule(static) reduction(+:sum_loss)
for
(
data_size_t
i
=
0
;
i
<
num_data_
;
++
i
)
{
// sigmoid transform
score_t
prob
=
1.0
f
/
(
1.0
f
+
std
::
exp
(
-
sigmoid_
*
score
[
i
]));
score_t
prob
=
1.0
f
/
(
1.0
f
+
std
::
exp
(
-
2.0
f
*
sigmoid_
*
score
[
i
]));
// add loss
sum_loss
+=
PointWiseLossCalculator
::
LossOnPoint
(
label_
[
i
],
prob
)
*
weights_
[
i
];
}
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment