PndMultiClassBdtTrain.h

Go to the documentation of this file.

/* ***************************************
 * MultiClass BDTG  Training functions   *
 * Author: M.Babai@rug.nl                *
 * Version:                              *
 * LICENSE:                              *
 * ***************************************
 */
/*
 * Note: This is just an interface to the original TMVA
 * implementation. To find out the available options, please read TMVA
 * manuals. In case of errors or wrong outputs produced by TMVA
 * classifiers, try to read their mailing list and send your questions
 * to the same list.
 ******* VERY IMORTANT ****
 * You NEED (TMVA version) > 4.1.X before this works.
 */
//#pragma once
#ifndef PND_MULTICLASS_BDT_TRAIN_H
#define PND_MULTICLASS_BDT_TRAIN_H

//Local includes
#include "PndMvaTrainer.h"

// TMVA && ROOT
#include "TMVA/Factory.h"
#include "TMVA/Config.h"

// Interface definition for Multiclass MLP trainers.
class PndMultiClassBdtTrain : public PndMvaTrainer
{
  //----------------------------------------
  //================== public ==============
 public:
  explicit PndMultiClassBdtTrain(std::string const& InPut,
                                 std::vector<std::string> const& ClassNames, 
                                 std::vector<std::string> const& VarNames,
                                 bool trim = true);
  virtual ~PndMultiClassBdtTrain();
  
  void Train();

  void storeWeights();

  void Initialize();

  //______________________________________________
  //====== Getters and setters.  
  // Set the name of the current job
  inline void SetJobName (std::string const& name);

  // Set data Transformation scheme
  inline void SetTransformation (std::string const& tran);
  
  // Set the options for the MLP alg. See TMVA manuals.
  inline void SetBdtOptions (std::string const& opts);
  
  // Set the file name to store evaluation outputs.
  inline void SetEvalFileName (std::string const& fname);
  
  // Set the directory where weights are stored.
  inline void SetWeightsOutDir (std::string const& dirName);
  
  // Evaluate the classifier?
  inline void SetEvaluation (bool evaluate);
  
  // Get the current job name.
  inline std::string const& GetJobName() const;
  
  // Get the current transformation info.
  inline std::string const& GetTransformation() const;
  
  // Get the classifier options.
  inline std::string const& GetBdtOptions() const;
  
  // Get the name of the weight file.
  inline std::string const& GetEvalFileName() const;
  
  // Get the directory where the weights are stored.
  inline std::string const& GetWeightsOutDir() const;
  //----------------------------------------
  
  //================== protected ============
  //protected:
  //----------------------------------------
  
  //================== private =============
 private:
  // To avoid mistakes.
  PndMultiClassBdtTrain(PndMultiClassBdtTrain const& oth);
  PndMultiClassBdtTrain& operator=(PndMultiClassBdtTrain const& oth);
  
  // Initialize mlp object and set the options.
  void InitBdt();
  // Add the variables to the TMVA factory object.
  void AddVariables();
  
  //==============================
  TMVA::Factory* m_factory;// TMVA factory
  TFile* EvalFile; // To store evaluation file
  std::string m_JName; //Job name
  std::string m_transform;// Transformation opt.
  std::string m_BdtOptions; // Bdt options.
  std::string m_evalFileName; //evaluation file name.
  std::string m_weightDirName;// Directory name to store weights.
  bool m_Evaluate;
};// End of interface definition.
//=============== inline functions implementation. ========
//__________________________________________
inline void PndMultiClassBdtTrain::SetJobName(std::string const& name)
{
  this->m_JName = name;
};

inline void PndMultiClassBdtTrain::SetTransformation(std::string const& tr)
{
  this->m_transform = tr;
};

inline void PndMultiClassBdtTrain::SetBdtOptions(std::string const& opt)
{
  this->m_BdtOptions = opt;
};

inline std::string const& PndMultiClassBdtTrain::GetJobName() const
{
  return m_JName;
};

inline std::string const& PndMultiClassBdtTrain::GetTransformation() const
{
  return m_transform;
};

inline std::string const& PndMultiClassBdtTrain::GetBdtOptions() const
{
  return m_BdtOptions;
};

inline void PndMultiClassBdtTrain::SetEvalFileName(std::string const& fname)
{
  this->m_evalFileName = fname;
};

inline std::string const& PndMultiClassBdtTrain::GetEvalFileName() const
{
  return m_evalFileName;
};

inline void PndMultiClassBdtTrain::SetWeightsOutDir (std::string const& dirName)
{
  this->m_weightDirName = dirName;
};

inline std::string const& PndMultiClassBdtTrain::GetWeightsOutDir() const
{
  return m_weightDirName;
};

inline void PndMultiClassBdtTrain::SetEvaluation(bool evaluate)
{
  this->m_Evaluate = evaluate;
};
#endif