TMVAHelper.h

Go to the documentation of this file.

#ifndef TMVAHelper_TMVAHelper_h
#define TMVAHelper_TMVAHelper_h
 
// ROOT
#include "ROOT/RVec.hxx"
#include "RVersion.h"
#include "TMVA/RBDT.hxx"
 
// TBB
#include <tbb/task_arena.h>
 
// std
#include <string>
#include <vector>
 
class tmva_helper_xgb {
public:
  tmva_helper_xgb(const std::string &filename, const std::string &name,
                  const unsigned int nslots = 1);
  ~tmva_helper_xgb() {};
  ROOT::VecOps::RVec<float> operator()(const ROOT::VecOps::RVec<float> vars);
 
private:
  // Default backend (template parameter) is:
  // TMVA::Experimental::BranchlessJittedForest<float>
#if ROOT_VERSION_CODE >= ROOT_VERSION(6, 32, 0)
  std::vector<TMVA::Experimental::RBDT> m_interpreters;
#else
  std::vector<TMVA::Experimental::RBDT<>> m_interpreters;
#endif
};
 
#endif