pass train dataset parser config to valid dataset loading parser

microsoft · Jan 25, 2022 · e380580 · e380580
1 parent 0075814
commit e380580
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/io/dataset_loader.cpp b/src/io/dataset_loader.cpp
@@ -291,7 +291,7 @@ Dataset* DatasetLoader::LoadFromFileAlignWithOtherDataset(const char* filename,
   auto bin_filename = CheckCanLoadFromBin(filename);
   if (bin_filename.size() == 0) {
     auto parser = std::unique_ptr<Parser>(Parser::CreateParser(filename, config_.header, 0, label_idx_,
-                                                               config_.precise_float_parser, dataset->parser_config_str_));
+                                                               config_.precise_float_parser, train_data->parser_config_str_));
     if (parser == nullptr) {
       Log::Fatal("Could not recognize data format of %s", filename);
     }