Unverified Commit 7d1276ad authored by Scott Votaw's avatar Scott Votaw Committed by GitHub
Browse files

Rename Metadata num_classes to be more clear (#5461)

Rename num_classes to be more clear
parent d78b6bc2
...@@ -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_);
} }
......
...@@ -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;
......
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