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
fe9061fa
"vscode:/vscode.git/clone" did not exist on "015c8fff7203b7266ce187938327bc7b478a7218"
Commit
fe9061fa
authored
Mar 07, 2017
by
Guolin Ke
Browse files
fix prediction in tree learner.
parent
07f709b9
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
0 deletions
+1
-0
src/treelearner/serial_tree_learner.h
src/treelearner/serial_tree_learner.h
+1
-0
No files found.
src/treelearner/serial_tree_learner.h
View file @
fe9061fa
...
...
@@ -43,6 +43,7 @@ public:
}
void
AddPredictionToScore
(
double
*
out_score
)
const
override
{
if
(
last_trained_tree_
->
num_leaves
()
<=
1
)
{
return
;
}
#pragma omp parallel for schedule(static)
for
(
int
i
=
0
;
i
<
data_partition_
->
num_leaves
();
++
i
)
{
double
output
=
static_cast
<
double
>
(
last_trained_tree_
->
LeafOutput
(
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