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
04ede377
Commit
04ede377
authored
Mar 27, 2017
by
zhangyafeikimi
Committed by
Qiwei Ye
Mar 27, 2017
Browse files
optimize performace (#363)
parent
d5bbaa26
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
8 additions
and
6 deletions
+8
-6
src/metric/multiclass_metric.hpp
src/metric/multiclass_metric.hpp
+2
-2
src/objective/multiclass_objective.hpp
src/objective/multiclass_objective.hpp
+6
-4
No files found.
src/metric/multiclass_metric.hpp
View file @
04ede377
...
...
@@ -100,7 +100,7 @@ class MultiErrorMetric: public MulticlassMetric<MultiErrorMetric> {
public:
explicit
MultiErrorMetric
(
const
MetricConfig
&
config
)
:
MulticlassMetric
<
MultiErrorMetric
>
(
config
)
{}
inline
static
double
LossOnPoint
(
float
label
,
std
::
vector
<
double
>
score
)
{
inline
static
double
LossOnPoint
(
float
label
,
std
::
vector
<
double
>
&
score
)
{
size_t
k
=
static_cast
<
size_t
>
(
label
);
for
(
size_t
i
=
0
;
i
<
score
.
size
();
++
i
){
if
(
i
!=
k
&&
score
[
i
]
>=
score
[
k
])
{
...
...
@@ -120,7 +120,7 @@ class MultiLoglossMetric: public MulticlassMetric<MultiLoglossMetric> {
public:
explicit
MultiLoglossMetric
(
const
MetricConfig
&
config
)
:
MulticlassMetric
<
MultiLoglossMetric
>
(
config
)
{}
inline
static
double
LossOnPoint
(
float
label
,
std
::
vector
<
double
>
score
)
{
inline
static
double
LossOnPoint
(
float
label
,
std
::
vector
<
double
>
&
score
)
{
size_t
k
=
static_cast
<
size_t
>
(
label
);
Common
::
Softmax
(
&
score
);
if
(
score
[
k
]
>
kEpsilon
)
{
...
...
src/objective/multiclass_objective.hpp
View file @
04ede377
...
...
@@ -48,9 +48,10 @@ public:
void
GetGradients
(
const
double
*
score
,
score_t
*
gradients
,
score_t
*
hessians
)
const
override
{
if
(
weights_
==
nullptr
)
{
#pragma omp parallel for schedule(static)
std
::
vector
<
double
>
rec
;
#pragma omp parallel for schedule(static) private(rec)
for
(
data_size_t
i
=
0
;
i
<
num_data_
;
++
i
)
{
std
::
vector
<
double
>
rec
(
num_class_
);
rec
.
resize
(
num_class_
);
for
(
int
k
=
0
;
k
<
num_class_
;
++
k
){
size_t
idx
=
static_cast
<
size_t
>
(
num_data_
)
*
k
+
i
;
rec
[
k
]
=
static_cast
<
double
>
(
score
[
idx
]);
...
...
@@ -69,9 +70,10 @@ public:
}
}
}
else
{
#pragma omp parallel for schedule(static)
std
::
vector
<
double
>
rec
;
#pragma omp parallel for schedule(static) private(rec)
for
(
data_size_t
i
=
0
;
i
<
num_data_
;
++
i
)
{
std
::
vector
<
double
>
rec
(
num_class_
);
rec
.
resize
(
num_class_
);
for
(
int
k
=
0
;
k
<
num_class_
;
++
k
){
size_t
idx
=
static_cast
<
size_t
>
(
num_data_
)
*
k
+
i
;
rec
[
k
]
=
static_cast
<
double
>
(
score
[
idx
]);
...
...
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