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

Method SetFloatField

src/io/dataset.cpp:543–562  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

541}
542
543bool Dataset::SetFloatField(const char* field_name, const float* field_data, data_size_t num_element) {
544 std::string name(field_name);
545 name = Common::Trim(name);
546 if (name == std::string("label") || name == std::string("target")) {
547 #ifdef LABEL_T_USE_DOUBLE
548 Log::Fatal("Don't support LABEL_T_USE_DOUBLE");
549 #else
550 metadata_.SetLabel(field_data, num_element);
551 #endif
552 } else if (name == std::string("weight") || name == std::string("weights")) {
553 #ifdef LABEL_T_USE_DOUBLE
554 Log::Fatal("Don't support LABEL_T_USE_DOUBLE");
555 #else
556 metadata_.SetWeights(field_data, num_element);
557 #endif
558 } else {
559 return false;
560 }
561 return true;
562}
563
564bool Dataset::SetDoubleField(const char* field_name, const double* field_data, data_size_t num_element) {
565 std::string name(field_name);

Callers 1

LGBM_DatasetSetFieldFunction · 0.45

Calls 3

TrimFunction · 0.50
SetLabelMethod · 0.45
SetWeightsMethod · 0.45

Tested by

no test coverage detected