DataFormats/BTauReco/interface/DeepDoubleXFeatures.h - master - githubmirror/cmssw

DataFormats/BTauReco/interface/DeepDoubleXFeatures.h

37 строк1 KB

Andrzej

feat: Introduce DeepDoubleX V2

30 июн 2020, 14:59

30 июн 2020, 14:592ddf272

#ifndef DataFormats_BTauReco_DeepDoubleXFeatures_h
#define DataFormats_BTauReco_DeepDoubleXFeatures_h

#include <vector>

#include "DataFormats/BTauReco/interface/JetFeatures.h"
#include "DataFormats/BTauReco/interface/SecondaryVertexFeatures.h"
#include "DataFormats/BTauReco/interface/BoostedDoubleSVTagInfoFeatures.h"
#include "DataFormats/BTauReco/interface/ChargedCandidateFeatures.h"
#include "DataFormats/BTauReco/interface/NeutralCandidateFeatures.h"

namespace btagbtvdeep {

  class DeepDoubleXFeatures {
  public:
    bool empty() const { return is_empty_; }

    void filled() { is_empty_ = false; }

    JetFeatures jet_features;
    BoostedDoubleSVTagInfoFeatures tag_info_features;

    std::vector<ChargedCandidateFeatures> c_pf_features;

    std::vector<NeutralCandidateFeatures> n_pf_features;

    std::vector<SecondaryVertexFeatures> sv_features;

    std::size_t npv;  // used by deep flavour

  private:
    bool is_empty_ = true;
  };

}  // namespace btagbtvdeep

#endif  //DataFormats_BTauReco_DeepDoubleXFeatures_h