File indexing completed on 2024-05-20 22:40:06
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 float rm_outlier(float val);
0010
0011 static const std::unordered_map<int, int32_t> charge_embedding{{-1, 0}, {0, 1}, {1, 2}};
0012 static const std::unordered_map<int, int32_t> pdg_id_embedding{
0013 {-211, 0}, {-13, 1}, {-11, 2}, {0, 3}, {1, 4}, {2, 5}, {11, 6}, {13, 7}, {22, 8}, {130, 9}, {211, 10}};
0014 }
0015
0016 #endif