PndMultiClassMlpClassify.h

Go to the documentation of this file.

/* ***************************************
 * 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_MLP_CLASSIFY_H
#define PND_MULTICLASS_MLP_CLASSIFY_H

//Local includes
#include "PndMvaClassifier.h"

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

// Interface definition for Multiclass MLP Classifier
class PndMultiClassMlpClassify : public PndMvaClassifier
{
  //______________ Public ____________________
 public:
  explicit PndMultiClassMlpClassify(std::string const& inputFile,
                                    std::vector<std::string> const& classNames, 
                                    std::vector<std::string> const& varNames);
  virtual ~PndMultiClassMlpClassify();
  
  void GetMvaValues( std::vector<float> EvtData,
                     std::map<std::string, float>& result );
  std::string *Classify( std::vector<float> EvtData );
  
  void Initialize();
  
  //______________________________________________
  //====== Getters and setters.
  // Set the TMVA reader options
  inline void SetReaderOptions(std::string const& opt);

  // Set the file name which contains the weights
  inline void SetWeightsFileName(std::string const& name);

  // Get the TMVA reader options
  inline std::string const& GetReaderOptions() const;

  // Get weights file name.
  inline std::string const& GetWeightsFileName() const;

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

  //======================
  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 PndMultiClassMlpClassify::SetReaderOptions(std::string const& opt)
{
  this->m_readerOptions = opt;
};

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

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

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