MCPcopy Create free account
hub / github.com/antmachineintelligence/mtgbmcode / LGBM_DatasetCreateFromFile

Function LGBM_DatasetCreateFromFile

src/c_api.cpp:478–501  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

476}
477
478int LGBM_DatasetCreateFromFile(const char* filename,
479 const char* parameters,
480 const DatasetHandle reference,
481 DatasetHandle* out) {
482 API_BEGIN();
483 auto param = Config::Str2Map(parameters);
484 Config config;
485 config.Set(param);
486 if (config.num_threads > 0) {
487 omp_set_num_threads(config.num_threads);
488 }
489 DatasetLoader loader(config, nullptr, 1, filename);
490 if (reference == nullptr) {
491 if (Network::num_machines() == 1) {
492 *out = loader.LoadFromFile(filename, "");
493 } else {
494 *out = loader.LoadFromFile(filename, "", Network::rank(), Network::num_machines());
495 }
496 } else {
497 *out = loader.LoadFromFileAlignWithOtherDataset(filename, "",
498 reinterpret_cast<const Dataset*>(reference));
499 }
500 API_END();
501}
502
503
504int LGBM_DatasetCreateFromSampledColumn(double** sample_data,

Callers 1

Calls 4

omp_set_num_threadsFunction · 0.50
SetMethod · 0.45
LoadFromFileMethod · 0.45

Tested by

no test coverage detected