#include <XGBooster.h>

Public Member Functions
void	addFeature (std::string name)

float	predict ()

void	reset ()
	Reset feature values. More...

void	set (std::string name, float value)

	XGBooster (std::string model_file)

	XGBooster (std::string model_file, std::string model_features)

Private Attributes
BoosterHandle	booster_

std::map< std::string, unsigned int >	feature_name_to_index_

std::vector< float >	features_

Detailed Description

Definition at line 11 of file XGBooster.h.

Constructor & Destructor Documentation

◆ XGBooster() [1/2]

XGBooster::XGBooster ( std::string model_file )

Definition at line 46 of file XGBooster.cc.

References booster_, and mps_update::status.

                                          {
   int status = XGBoosterCreate(nullptr, 0, &booster_);
   if (status != 0)
     throw std::runtime_error("Failed to create XGBooster");
   status = XGBoosterLoadModel(booster_, model_file.c_str());
   if (status != 0)
     throw std::runtime_error("Failed to load XGBoost model");
   XGBoosterSetParam(booster_, "nthread", "1");
 }

◆ XGBooster() [2/2]

XGBooster::XGBooster	(	std::string	model_file,
		std::string	model_features
	)

Definition at line 56 of file XGBooster.cc.

References addFeature(), Skims_PA_cff::content, lowptgsfeleseed::features(), geometryDiff::file, read_features(), and AlCaHLTBitMon_QueryRunRegistry::string.

                                                                    : XGBooster(model_file) {
   std::ifstream file(model_features);
   if (!file.is_open())
     throw std::runtime_error("Failed to open file: " + model_features);
 
   std::string content((std::istreambuf_iterator<char>(file)), std::istreambuf_iterator<char>());
   file.close();
 
   std::vector<std::string> features = read_features(content);
 
   for (const auto& feature : features) {
     addFeature(feature);
   }
 }

Member Function Documentation

◆ addFeature()

void XGBooster::addFeature ( std::string name )

Features need to be entered in the order they are used in the model

Definition at line 73 of file XGBooster.cc.

References feature_name_to_index_, features_, and Skims_PA_cff::name.

Referenced by XGBooster().

                                          {
   features_.push_back(0);
   feature_name_to_index_[name] = features_.size() - 1;
 }

◆ predict()

float XGBooster::predict ( )

Definition at line 80 of file XGBooster.cc.

References cms::cuda::assert(), booster_, feature_name_to_index_, features_, mps_fire::i, CommonMethods::isnan(), reset(), mps_fire::result, runTheMatrix::ret, offlineSlimmedPrimaryVertices_cfi::score, and AlCaHLTBitMon_QueryRunRegistry::string.

Referenced by pat::computeSoftMvaRun3().

                          {
   float result(-999.);
 
   // check if all feature values are set properly
   for (unsigned int i = 0; i < features_.size(); ++i)
     if (std::isnan(features_.at(i))) {
       std::string feature_name;
       for (const auto& pair : feature_name_to_index_) {
         if (pair.second == i) {
           feature_name = pair.first;
           break;
         }
       }
       throw std::runtime_error("Feature is not set: " + feature_name);
     }
 
   DMatrixHandle dvalues;
   XGDMatrixCreateFromMat(&features_[0], 1, features_.size(), 9e99, &dvalues);
 
   bst_ulong out_len = 0;
   const float* score = nullptr;
 
   // config json
   const char* json = R"({
     "type": 0,
     "training": false,
     "iteration_begin": 0,
     "iteration_end": 0,
     "strict_shape": false
    })";
 
   // Shape of output prediction
   bst_ulong const* out_shape = nullptr;
 
   auto ret = XGBoosterPredictFromDMatrix(booster_, dvalues, json, &out_shape, &out_len, &score);
 
   XGDMatrixFree(dvalues);
 
   if (ret == 0) {
     assert(out_len == 1 && "Unexpected prediction format");
     result = score[0];
   }
 
   reset();
 
   return result;
 }