FastBDTOptions Class Reference

Options for the FANN MVA method. More...

#include <FastBDT.h>

Inheritance diagram for FastBDTOptions:

Collaboration diagram for FastBDTOptions:

Public Member Functions
virtual void	load (const boost::property_tree::ptree &pt) override
	Load mechanism to load Options from a xml tree. More...
virtual void	save (boost::property_tree::ptree &pt) const override
	Save mechanism to store Options in a xml tree. More...
virtual po::options_description	getDescription () override
	Returns a program options description for all available options.
virtual std::string	getMethod () const override
	Return method name.

Public Attributes
unsigned int	m_nTrees = 200
	Number of trees.
unsigned int	m_nCuts = 8
	Number of cut Levels = log_2(Number of Cuts)
unsigned int	m_nLevels = 3
	Depth of tree.
double	m_shrinkage = 0.1
	Shrinkage during the boosting step.
double	m_randRatio = 0.5
	Fraction of data to use in the stochastic training.

Detailed Description

Options for the FANN MVA method.

Definition at line 53 of file FastBDT.h.

Member Function Documentation

load()

void load ( const boost::property_tree::ptree &  pt )

overridevirtual

Load mechanism to load Options from a xml tree.

Parameters

pt

xml tree

Implements Options.

Definition at line 53 of file FastBDT.cc.

    {
      int version = pt.get<int>("FastBDT_version");
#if FastBDT_VERSION_MAJOR >= 5
      if (version != 1 and version != 2) {
        B2ERROR("Unknown weightfile version " << std::to_string(version));
        throw std::runtime_error("Unknown weightfile version " + std::to_string(version));
      }
#else
      if (version != 1) {
        B2ERROR("Unknown weightfile version " << std::to_string(version));
        throw std::runtime_error("Unknown weightfile version " + std::to_string(version));
      }
#endif
      m_nTrees = pt.get<int>("FastBDT_nTrees");
      m_nCuts = pt.get<int>("FastBDT_nCuts");
      m_nLevels = pt.get<int>("FastBDT_nLevels");
      m_shrinkage = pt.get<double>("FastBDT_shrinkage");
      m_randRatio = pt.get<double>("FastBDT_randRatio");
 
#if FastBDT_VERSION_MAJOR >= 5
      if (version > 1) {
 
        m_flatnessLoss = pt.get<double>("FastBDT_flatnessLoss");
        m_sPlot = pt.get<bool>("FastBDT_sPlot");
 
        unsigned int numberOfIndividualNCuts = pt.get<unsigned int>("FastBDT_number_individual_nCuts", 0);
        m_individual_nCuts.resize(numberOfIndividualNCuts);
        for (unsigned int i = 0; i < numberOfIndividualNCuts; ++i) {
          m_individual_nCuts[i] = pt.get<unsigned int>(std::string("FastBDT_individual_nCuts") + std::to_string(i));
        }
 
        m_purityTransformation = pt.get<bool>("FastBDT_purityTransformation");
        unsigned int numberOfIndividualPurityTransformation = pt.get<unsigned int>("FastBDT_number_individualPurityTransformation", 0);
        m_individualPurityTransformation.resize(numberOfIndividualPurityTransformation);
        for (unsigned int i = 0; i < numberOfIndividualPurityTransformation; ++i) {
          m_individualPurityTransformation[i] = pt.get<bool>(std::string("FastBDT_individualPurityTransformation") + std::to_string(i));
        }
 
      } else {
        m_flatnessLoss = -1.0;
        m_sPlot = false;
      }
#endif
    }

save()

void save ( boost::property_tree::ptree &  pt ) const

overridevirtual

Save mechanism to store Options in a xml tree.

Parameters

pt

xml tree

Implements Options.

Definition at line 99 of file FastBDT.cc.

---

The documentation for this class was generated from the following files:

• mva/methods/include/FastBDT.h
• mva/methods/src/FastBDT.cc