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
7d1276ad
Unverified
Commit
7d1276ad
authored
Sep 01, 2022
by
Scott Votaw
Committed by
GitHub
Sep 02, 2022
Browse files
Rename Metadata num_classes to be more clear (#5461)
Rename num_classes to be more clear
parent
d78b6bc2
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
4 additions
and
4 deletions
+4
-4
include/LightGBM/dataset.h
include/LightGBM/dataset.h
+2
-2
src/io/metadata.cpp
src/io/metadata.cpp
+2
-2
No files found.
include/LightGBM/dataset.h
View file @
7d1276ad
...
@@ -161,7 +161,7 @@ class Metadata {
...
@@ -161,7 +161,7 @@ class Metadata {
* \param values Initial score values for this record, one per class
* \param values Initial score values for this record, one per class
*/
*/
inline
void
SetInitScoreAt
(
data_size_t
idx
,
const
double
*
values
)
{
inline
void
SetInitScoreAt
(
data_size_t
idx
,
const
double
*
values
)
{
const
auto
nclasses
=
num_classes
();
const
auto
nclasses
=
num_
init_score_
classes
();
const
double
*
val_ptr
=
values
;
const
double
*
val_ptr
=
values
;
for
(
int
i
=
idx
;
i
<
nclasses
*
num_data_
;
i
+=
num_data_
,
++
val_ptr
)
{
for
(
int
i
=
idx
;
i
<
nclasses
*
num_data_
;
i
+=
num_data_
,
++
val_ptr
)
{
init_score_
[
i
]
=
*
val_ptr
;
init_score_
[
i
]
=
*
val_ptr
;
...
@@ -265,7 +265,7 @@ class Metadata {
...
@@ -265,7 +265,7 @@ class Metadata {
/*!
/*!
* \brief Get number of classes
* \brief Get number of classes
*/
*/
inline
int32_t
num_classes
()
const
{
inline
int32_t
num_
init_score_
classes
()
const
{
if
(
num_data_
&&
num_init_score_
)
{
if
(
num_data_
&&
num_init_score_
)
{
return
static_cast
<
int
>
(
num_init_score_
/
num_data_
);
return
static_cast
<
int
>
(
num_init_score_
/
num_data_
);
}
}
...
...
src/io/metadata.cpp
View file @
7d1276ad
...
@@ -62,7 +62,7 @@ void Metadata::InitByReference(data_size_t num_data, const Metadata* reference)
...
@@ -62,7 +62,7 @@ void Metadata::InitByReference(data_size_t num_data, const Metadata* reference)
int
has_weights
=
reference
->
num_weights_
>
0
;
int
has_weights
=
reference
->
num_weights_
>
0
;
int
has_init_scores
=
reference
->
num_init_score_
>
0
;
int
has_init_scores
=
reference
->
num_init_score_
>
0
;
int
has_queries
=
reference
->
num_queries_
>
0
;
int
has_queries
=
reference
->
num_queries_
>
0
;
int
nclasses
=
reference
->
num_classes
();
int
nclasses
=
reference
->
num_
init_score_
classes
();
Init
(
num_data
,
has_weights
,
has_init_scores
,
has_queries
,
nclasses
);
Init
(
num_data
,
has_weights
,
has_init_scores
,
has_queries
,
nclasses
);
}
}
...
@@ -361,7 +361,7 @@ void Metadata::InsertInitScores(const double* init_scores, data_size_t start_ind
...
@@ -361,7 +361,7 @@ void Metadata::InsertInitScores(const double* init_scores, data_size_t start_ind
}
}
if
(
init_score_
.
empty
())
{
init_score_
.
resize
(
num_init_score_
);
}
if
(
init_score_
.
empty
())
{
init_score_
.
resize
(
num_init_score_
);
}
int
nclasses
=
num_classes
();
int
nclasses
=
num_
init_score_
classes
();
for
(
int32_t
col
=
0
;
col
<
nclasses
;
++
col
)
{
for
(
int32_t
col
=
0
;
col
<
nclasses
;
++
col
)
{
int32_t
dest_offset
=
num_data_
*
col
+
start_index
;
int32_t
dest_offset
=
num_data_
*
col
+
start_index
;
...
...
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