diff --git a/src/io/dataset_loader.cpp b/src/io/dataset_loader.cpp index 8350520d5369..3caa8bb31a16 100644 --- a/src/io/dataset_loader.cpp +++ b/src/io/dataset_loader.cpp @@ -250,6 +250,11 @@ Dataset* DatasetLoader::LoadFromFile(const char* filename, int rank, int num_mac } // check meta data dataset->metadata_.CheckOrPartition(num_global_data, used_data_indices); + data_size_t num_queries = dataset->metadata_.num_queries(); + if (num_queries > 0) { + Log::Debug("Number of queries in %s: %i. Average number of rows per query: %f.", + filename, static_cast(num_queries), static_cast(dataset->num_data_) / num_queries); + } // need to check training data CheckDataset(dataset.get(), is_load_from_binary); diff --git a/src/io/metadata.cpp b/src/io/metadata.cpp index 859a67c77ca3..63a1690906a2 100644 --- a/src/io/metadata.cpp +++ b/src/io/metadata.cpp @@ -162,8 +162,6 @@ void Metadata::CheckOrPartition(data_size_t num_all_data, const std::vector(num_queries_), static_cast(num_data_) / num_queries_); LoadQueryWeights(); queries_.clear(); }