PndMultiClassBdtClassify.h

Go to the documentation of this file.

//****************************************************************************
//*                   This file is part of PandaRoot.                        *
//*                                                                          *
//*            PandaRoot is distributed under the terms of the               *
//*              GNU General Public License (GPL) version 3,                 *
//*                 copied verbatim in the file "LICENSE".                   *
//*                                                                          *
//*  Copyright (C) 2006 - 2024 FAIR GmbH and copyright holders of PandaRoot  *
//*     The copyright holders are listed in the file "COPYRIGHTHOLDERS".     *
//*               The authors are listed in the file "AUTHORS".              *
//****************************************************************************

/* ***************************************
 * MultiClass MLP Classifier 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_CLASSIFY_H
#define PND_MULTICLASS_BDT_CLASSIFY_H

// Local includes
#include "PndMvaClassifier.h"

// Root && TMVA
#include "TMVA/Reader.h"

// Interface definition for Multiclass MLP Classifier
class PndMultiClassBdtClassify : public PndMvaClassifier {
  //______________ Public ____________________
 public:
  explicit PndMultiClassBdtClassify(std::string const &inputFile, std::vector<std::string> const &classNames, std::vector<std::string> const &varNames);
  virtual ~PndMultiClassBdtClassify();

  void GetMvaValues(std::vector<float> EvtData, std::map<std::string, float> &result);
  std::string *Classify(std::vector<float> EvtData);

  void Initialize();

  //______________________________________________
  //====== Getters and setters.
  inline void SetReaderOptions(std::string const &opt);
  inline void SetWeightsFileName(std::string const &name);

  inline std::string const &GetReaderOptions() const;
  inline std::string const &GetWeightsFileName() const;

  //______________ Protected__________________
  // protected:
  //______________ Private ____________________
 private:
  // To avoid mistakes.
  PndMultiClassBdtClassify(PndMultiClassBdtClassify const &oth);
  PndMultiClassBdtClassify &operator=(PndMultiClassBdtClassify const &oth);

  // Set the TMVA reader options.
  void SetOptions();

  //======================
  TMVA::Reader *m_reader; 

  std::string m_weightsFile;           
  std::string m_readerOptions;         
  std::vector<float> m_EventContainer; 
};                                     // End of classifier interface definition.

//=============== inline functions implementation. ========
//__________________________________________
inline void PndMultiClassBdtClassify::SetReaderOptions(std::string const &opt)
{
  this->m_readerOptions = opt;
};

inline std::string const &PndMultiClassBdtClassify::GetReaderOptions() const
{
  return m_readerOptions;
};

inline void PndMultiClassBdtClassify::SetWeightsFileName(std::string const &name)
{
  this->m_weightsFile = name;
};

inline std::string const &PndMultiClassBdtClassify::GetWeightsFileName() const
{
  return m_weightsFile;
};
#endif