Back to home page

Project CMSSW displayed by LXR

 
 

    


File indexing completed on 2025-01-31 02:19:53

0001 #ifndef RecoBTau_JetTagComputer_TagInfoMVACategorySelector_h
0002 #define RecoBTau_JetTagComputer_TagInfoMVACategorySelector_h
0003 
0004 #include <string>
0005 #include <vector>
0006 
0007 #include "DataFormats/BTauReco/interface/TaggingVariable.h"
0008 #include "FWCore/ParameterSet/interface/ConfigurationDescriptions.h"
0009 #include "FWCore/ParameterSet/interface/ParameterSet.h"
0010 #include "FWCore/ParameterSet/interface/ParameterSetDescription.h"
0011 #include "PhysicsTools/MVAComputer/interface/Calibration.h"
0012 
0013 class TagInfoMVACategorySelector {
0014 public:
0015   TagInfoMVACategorySelector(const edm::ParameterSet &params);
0016   ~TagInfoMVACategorySelector();
0017 
0018   inline const std::vector<std::string> &getCategoryLabels() const { return categoryLabels; }
0019 
0020   int findCategory(const reco::TaggingVariableList &taggingVariables) const;
0021 
0022   static void fillPSetDescription(edm::ParameterSetDescription &desc);
0023 
0024 private:
0025   reco::TaggingVariableName categoryVariable;
0026   std::vector<std::string> categoryLabels;
0027 };
0028 
0029 #endif  // RecoBTau_JetTagComputer_TagInfoMVACategorySelector_h