ForestBase.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2019 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef MVAUtils_ForestBase_H
#define MVAUtils_ForestBase_H
 
#include <vector>
 
class TTree;
class TString;
 
namespace MVAUtils
{
 
    class IForest
    {
    public:
        virtual float GetTreeResponse(const std::vector<float>& values, unsigned int itree) const = 0;
        virtual float GetTreeResponse(const std::vector<float*>& pointers, unsigned int itree) const = 0;
        virtual float GetOffset() const = 0;
        virtual float GetRawResponse(const std::vector<float>& values) const = 0;
        virtual float GetRawResponse(const std::vector<float*>& pointers) const = 0;
        virtual float GetClassification(const std::vector<float>& values) const = 0;
        virtual float GetClassification(const std::vector<float*>& pointers) const = 0;
        virtual float GetResponse(const std::vector<float>& values) const = 0;
        virtual float GetResponse(const std::vector<float*>& pointers) const = 0;
        virtual std::vector<float> GetMultiResponse(const std::vector<float>& values,
                                                    unsigned int numClasses) const = 0;
        virtual std::vector<float>  GetMultiResponse(const std::vector<float*>& pointers,
                                                     unsigned int numClasses) const = 0;
        virtual unsigned int GetNTrees() const = 0;
        virtual void PrintForest() const = 0;
        virtual void PrintTree(unsigned int itree) const = 0;
        virtual TTree* WriteTree(TString) const = 0;
        virtual int GetNVars() const = 0;
        virtual ~IForest()=default;
    };
 
}
#endif