METPUSubtraction/interface/DeepMETHelp.h

0001 #ifndef RecoMET_METPUSubtraction_DeepMETHelp_h
0002 #define RecoMET_METPUSubtraction_DeepMETHelp_h
0003
0004 #include <unordered_map>
0005 #include <cstdint>
0006
0007 namespace deepmet_helper {
0008   float scale_and_rm_outlier(float val, float scale);
0009
0010   static const std::unordered_map<int, int32_t> charge_embedding{{-1, 0}, {0, 1}, {1, 2}};
0011   static const std::unordered_map<int, int32_t> pdg_id_embedding{
0012       {-211, 0}, {-13, 1}, {-11, 2}, {0, 3}, {1, 4}, {2, 5}, {11, 6}, {13, 7}, {22, 8}, {130, 9}, {211, 10}};
0013 }  // namespace deepmet_helper
0014
0015 #endif