File indexing completed on 2023-03-17 11:20:13
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 }
0014
0015 #endif