Skip to content

Commit 3981b34

Browse files
Change debug logging location so that we can print the data file name as well.
1 parent af48add commit 3981b34

File tree

2 files changed

+5
-2
lines changed

2 files changed

+5
-2
lines changed

src/io/dataset_loader.cpp

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -250,6 +250,11 @@ Dataset* DatasetLoader::LoadFromFile(const char* filename, int rank, int num_mac
250250
}
251251
// check meta data
252252
dataset->metadata_.CheckOrPartition(num_global_data, used_data_indices);
253+
data_size_t num_queries = dataset->metadata_.num_queries();
254+
if (num_queries > 0) {
255+
Log::Debug("Number of queries in %s: %i. Average number of rows per query: %f.",
256+
filename, static_cast<int>(num_queries), static_cast<double>(dataset->num_data_) / num_queries);
257+
}
253258
// need to check training data
254259
CheckDataset(dataset.get(), is_load_from_binary);
255260

src/io/metadata.cpp

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -162,8 +162,6 @@ void Metadata::CheckOrPartition(data_size_t num_all_data, const std::vector<data
162162
for (size_t i = 0; i < tmp_buffer.size(); ++i) {
163163
query_boundaries_[i + 1] = query_boundaries_[i] + tmp_buffer[i];
164164
}
165-
Log::Info("Number of queries: %i. Average number of rows per query: %f.",
166-
static_cast<int>(num_queries_), static_cast<double>(num_data_) / num_queries_);
167165
LoadQueryWeights();
168166
queries_.clear();
169167
}

0 commit comments

Comments
 (0)