QuickMVAProbEstimator.h

Go to the documentation of this file.

#ifndef ROOT_TMVA_QUICKMVAPROBESTIMATOR 
#define ROOT_TMVA_QUICKMVAPROBESTIMATOR 

#include <iostream>
#include <vector>
#include <algorithm>

#include "TMVA/MsgLogger.h"

namespace TMVA {

  class QuickMVAProbEstimator {
  public:
    
    struct EventInfo{
      Double_t eventValue;
      Double_t eventWeight; 
      Int_t    eventType;  //signal or background
    };
    static bool compare(EventInfo e1, EventInfo e2){return e1.eventValue < e2.eventValue;}
    
    QuickMVAProbEstimator(Int_t nMin=40, Int_t nMax=5000):fIsSorted(false),fNMin(nMin),fNMax(nMax){ fLogger = new MsgLogger("QuickMVAProbEstimator");}
    
    
    virtual ~QuickMVAProbEstimator(){delete fLogger;}
    void AddEvent(Double_t val, Double_t weight, Int_t type);
    
   
    Double_t GetMVAProbAt(Double_t value);
    
    
  private:
    std::vector<EventInfo> fEvtVector;
    Bool_t                 fIsSorted;
    UInt_t                 fNMin;
    UInt_t                 fNMax;
    
    mutable MsgLogger*    fLogger;
    MsgLogger& Log() const { return *fLogger; }
    
    ClassDef(QuickMVAProbEstimator,0) // Interface to different separation critiera used in training algorithms
      
      
  };
}


#endif