diff --git a/ParticleNetAK8/General/V01/README.md b/ParticleNetAK8/General/V01/README.md new file mode 100644 index 0000000..1172a9f --- /dev/null +++ b/ParticleNetAK8/General/V01/README.md @@ -0,0 +1,5 @@ +# ParticleNetAK8-General-V01 + +This folder contains the ONNX models for the ParticleNet tagger designed for identifying Top/W/Z/H and their decay modes. No mass decorrelation is performed. The model is trained on AK8Puppi jets using `RunIISummer19UL17MiniAOD` and `RunIISummer19UL18MiniAOD` samples with Puppi tune `V14`. + +The output classes of the DNN can be found in the `preprocess.json` file. diff --git a/ParticleNetAK8/General/V01/particle-net.onnx b/ParticleNetAK8/General/V01/particle-net.onnx new file mode 100755 index 0000000..fd94d4b Binary files /dev/null and b/ParticleNetAK8/General/V01/particle-net.onnx differ diff --git a/ParticleNetAK8/General/V01/preprocess.json b/ParticleNetAK8/General/V01/preprocess.json new file mode 100755 index 0000000..7dc651e --- /dev/null +++ b/ParticleNetAK8/General/V01/preprocess.json @@ -0,0 +1,458 @@ +{ + "output_names": [ + "label_Top_bcq", + "label_Top_bqq", + "label_Top_bc", + "label_Top_bq", + "label_Top_bele", + "label_Top_bmu", + "label_Top_btau", + "label_W_cq", + "label_W_qq", + "label_Z_bb", + "label_Z_cc", + "label_Z_qq", + "label_H_bb", + "label_H_cc", + "label_H_qqqq", + "label_QCD_bb", + "label_QCD_cc", + "label_QCD_b", + "label_QCD_c", + "label_QCD_others" + ], + "input_names": [ + "pf_points", + "pf_features", + "pf_mask", + "sv_points", + "sv_features", + "sv_mask" + ], + "pf_points": { + "var_names": [ + "pfcand_etarel", + "pfcand_phirel" + ], + "var_infos": { + "pfcand_etarel": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "pfcand_phirel": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + } + }, + "min_length": 16, + "max_length": 100 + }, + "pf_features": { + "var_names": [ + "pfcand_pt_log_nopuppi", + "pfcand_e_log_nopuppi", + "pfcand_etarel", + "pfcand_phirel", + "pfcand_abseta", + "pfcand_charge", + "pfcand_isEl", + "pfcand_isMu", + "pfcand_isGamma", + "pfcand_isChargedHad", + "pfcand_isNeutralHad", + "pfcand_VTX_ass", + "pfcand_lostInnerHits", + "pfcand_normchi2", + "pfcand_quality", + "pfcand_dz", + "pfcand_dzsig", + "pfcand_dxy", + "pfcand_dxysig", + "pfcand_btagEtaRel", + "pfcand_btagPtRatio", + "pfcand_btagPParRatio", + "pfcand_btagSip3dVal", + "pfcand_btagSip3dSig", + "pfcand_btagJetDistVal" + ], + "var_infos": { + "pfcand_pt_log_nopuppi": { + "median": 1, + "norm_factor": 0.5, + "replace_inf_value": 0, + "lower_bound": -5, + "upper_bound": 5, + "pad": 0 + }, + "pfcand_e_log_nopuppi": { + "median": 1.3, + "norm_factor": 0.5, + "replace_inf_value": 0, + "lower_bound": -5, + "upper_bound": 5, + "pad": 0 + }, + "pfcand_etarel": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "pfcand_phirel": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "pfcand_abseta": { + "median": 0.6, + "norm_factor": 1.6, + "replace_inf_value": 0, + "lower_bound": -5, + "upper_bound": 5, + "pad": 0 + }, + "pfcand_charge": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "pfcand_isEl": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "pfcand_isMu": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "pfcand_isGamma": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "pfcand_isChargedHad": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "pfcand_isNeutralHad": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "pfcand_VTX_ass": { + "median": 4, + "norm_factor": 0.3, + "replace_inf_value": 0, + "lower_bound": -5, + "upper_bound": 5, + "pad": 0 + }, + "pfcand_lostInnerHits": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "pfcand_normchi2": { + "median": 5, + "norm_factor": 0.2, + "replace_inf_value": 0, + "lower_bound": -5, + "upper_bound": 5, + "pad": 0 + }, + "pfcand_quality": { + "median": 0, + "norm_factor": 0.2, + "replace_inf_value": 0, + "lower_bound": -5, + "upper_bound": 5, + "pad": 0 + }, + "pfcand_dz": { + "median": 0, + "norm_factor": 180, + "replace_inf_value": 0, + "lower_bound": -5, + "upper_bound": 5, + "pad": 0 + }, + "pfcand_dzsig": { + "median": 0, + "norm_factor": 0.9, + "replace_inf_value": 0, + "lower_bound": -5, + "upper_bound": 5, + "pad": 0 + }, + "pfcand_dxy": { + "median": 0.0, + "norm_factor": 300, + "replace_inf_value": 0, + "lower_bound": -5, + "upper_bound": 5, + "pad": 0 + }, + "pfcand_dxysig": { + "median": 0, + "norm_factor": 1.0, + "replace_inf_value": 0, + "lower_bound": -5, + "upper_bound": 5, + "pad": 0 + }, + "pfcand_btagEtaRel": { + "median": 1.5, + "norm_factor": 0.5, + "replace_inf_value": 0, + "lower_bound": -5, + "upper_bound": 5, + "pad": 0 + }, + "pfcand_btagPtRatio": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -5, + "upper_bound": 5, + "pad": 0 + }, + "pfcand_btagPParRatio": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -5, + "upper_bound": 5, + "pad": 0 + }, + "pfcand_btagSip3dVal": { + "median": 0, + "norm_factor": 100, + "replace_inf_value": 0, + "lower_bound": -5, + "upper_bound": 5, + "pad": 0 + }, + "pfcand_btagSip3dSig": { + "median": 0, + "norm_factor": 0.5, + "replace_inf_value": 0, + "lower_bound": -5, + "upper_bound": 5, + "pad": 0 + }, + "pfcand_btagJetDistVal": { + "median": 0, + "norm_factor": 40, + "replace_inf_value": 0, + "lower_bound": -5, + "upper_bound": 5, + "pad": 0 + } + }, + "min_length": 16, + "max_length": 100 + }, + "pf_mask": { + "var_names": [ + "pfcand_mask" + ], + "var_infos": { + "pfcand_mask": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + } + }, + "min_length": 16, + "max_length": 100 + }, + "sv_points": { + "var_names": [ + "sv_etarel", + "sv_phirel" + ], + "var_infos": { + "sv_etarel": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "sv_phirel": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + } + }, + "min_length": 1, + "max_length": 10 + }, + "sv_features": { + "var_names": [ + "sv_pt_log", + "sv_mass", + "sv_etarel", + "sv_phirel", + "sv_abseta", + "sv_ntracks", + "sv_normchi2", + "sv_dxy", + "sv_dxysig", + "sv_d3d", + "sv_d3dsig" + ], + "var_infos": { + "sv_pt_log": { + "median": 4, + "norm_factor": 0.6, + "replace_inf_value": 0, + "lower_bound": -5, + "upper_bound": 5, + "pad": 0 + }, + "sv_mass": { + "median": 1.2, + "norm_factor": 0.3, + "replace_inf_value": 0, + "lower_bound": -5, + "upper_bound": 5, + "pad": 0 + }, + "sv_etarel": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "sv_phirel": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "sv_abseta": { + "median": 0.5, + "norm_factor": 1.6, + "replace_inf_value": 0, + "lower_bound": -5, + "upper_bound": 5, + "pad": 0 + }, + "sv_ntracks": { + "median": 3, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -5, + "upper_bound": 5, + "pad": 0 + }, + "sv_normchi2": { + "median": 0.8, + "norm_factor": 0.6, + "replace_inf_value": 0, + "lower_bound": -5, + "upper_bound": 5, + "pad": 0 + }, + "sv_dxy": { + "median": 0.4, + "norm_factor": 0.25, + "replace_inf_value": 0, + "lower_bound": -5, + "upper_bound": 5, + "pad": 0 + }, + "sv_dxysig": { + "median": 7, + "norm_factor": 0.02, + "replace_inf_value": 0, + "lower_bound": -5, + "upper_bound": 5, + "pad": 0 + }, + "sv_d3d": { + "median": 0.5, + "norm_factor": 0.2, + "replace_inf_value": 0, + "lower_bound": -5, + "upper_bound": 5, + "pad": 0 + }, + "sv_d3dsig": { + "median": 7, + "norm_factor": 0.02, + "replace_inf_value": 0, + "lower_bound": -5, + "upper_bound": 5, + "pad": 0 + } + }, + "min_length": 1, + "max_length": 10 + }, + "sv_mask": { + "var_names": [ + "sv_mask" + ], + "var_infos": { + "sv_mask": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + } + }, + "min_length": 1, + "max_length": 10 + } +} \ No newline at end of file diff --git a/ParticleNetAK8/MD-2prong/V01/README.md b/ParticleNetAK8/MD-2prong/V01/README.md new file mode 100644 index 0000000..e846f03 --- /dev/null +++ b/ParticleNetAK8/MD-2prong/V01/README.md @@ -0,0 +1,5 @@ +# ParticleNetAK8-MD-2prong-V01 + +This folder contains the ONNX models for the ParticleNet tagger designed for mass-independent 2-prong jet tagging (i.e., X->bb, X->cc, X->qq). The mass decorrelation is achieved by using a dedicated signal sample w/ a flat mass distribution (`BulkGravitonToHHTo4Q_MX-600to6000_MH-15to250_*`). The model is trained on AK8Puppi jets using `RunIISummer19UL17MiniAOD` and `RunIISummer19UL18MiniAOD` samples with Puppi tune `V14`. + +The output classes of the DNN can be found in the `preprocess.json` file. diff --git a/ParticleNetAK8/MD-2prong/V01/particle-net.onnx b/ParticleNetAK8/MD-2prong/V01/particle-net.onnx new file mode 100755 index 0000000..0edbf92 Binary files /dev/null and b/ParticleNetAK8/MD-2prong/V01/particle-net.onnx differ diff --git a/ParticleNetAK8/MD-2prong/V01/preprocess.json b/ParticleNetAK8/MD-2prong/V01/preprocess.json new file mode 100755 index 0000000..cfe4c57 --- /dev/null +++ b/ParticleNetAK8/MD-2prong/V01/preprocess.json @@ -0,0 +1,401 @@ +{ + "output_names": [ + "label_H_bb", + "label_H_cc", + "label_H_qq", + "label_QCD_bb", + "label_QCD_cc", + "label_QCD_b", + "label_QCD_c", + "label_QCD_others" + ], + "input_names": [ + "pf_points", + "pf_features", + "pf_mask", + "sv_points", + "sv_features", + "sv_mask" + ], + "pf_points": { + "var_names": [ + "pfcand_etarel", + "pfcand_phirel" + ], + "var_infos": { + "pfcand_etarel": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "pfcand_phirel": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + } + }, + "min_length": 16, + "max_length": 100 + }, + "pf_features": { + "var_names": [ + "pfcand_pt_log_nopuppi", + "pfcand_e_log_nopuppi", + "pfcand_etarel", + "pfcand_phirel", + "pfcand_abseta", + "pfcand_charge", + "pfcand_VTX_ass", + "pfcand_lostInnerHits", + "pfcand_normchi2", + "pfcand_quality", + "pfcand_dz", + "pfcand_dzsig", + "pfcand_dxy", + "pfcand_dxysig", + "pfcand_btagEtaRel", + "pfcand_btagPtRatio", + "pfcand_btagPParRatio", + "pfcand_btagSip3dVal", + "pfcand_btagSip3dSig", + "pfcand_btagJetDistVal" + ], + "var_infos": { + "pfcand_pt_log_nopuppi": { + "median": 1, + "norm_factor": 0.5, + "replace_inf_value": 0, + "lower_bound": -5, + "upper_bound": 5, + "pad": 0 + }, + "pfcand_e_log_nopuppi": { + "median": 1.3, + "norm_factor": 0.5, + "replace_inf_value": 0, + "lower_bound": -5, + "upper_bound": 5, + "pad": 0 + }, + "pfcand_etarel": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "pfcand_phirel": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "pfcand_abseta": { + "median": 0.6, + "norm_factor": 1.6, + "replace_inf_value": 0, + "lower_bound": -5, + "upper_bound": 5, + "pad": 0 + }, + "pfcand_charge": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "pfcand_VTX_ass": { + "median": 4, + "norm_factor": 0.3, + "replace_inf_value": 0, + "lower_bound": -5, + "upper_bound": 5, + "pad": 0 + }, + "pfcand_lostInnerHits": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "pfcand_normchi2": { + "median": 5, + "norm_factor": 0.2, + "replace_inf_value": 0, + "lower_bound": -5, + "upper_bound": 5, + "pad": 0 + }, + "pfcand_quality": { + "median": 0, + "norm_factor": 0.2, + "replace_inf_value": 0, + "lower_bound": -5, + "upper_bound": 5, + "pad": 0 + }, + "pfcand_dz": { + "median": 0, + "norm_factor": 180, + "replace_inf_value": 0, + "lower_bound": -5, + "upper_bound": 5, + "pad": 0 + }, + "pfcand_dzsig": { + "median": 0, + "norm_factor": 0.9, + "replace_inf_value": 0, + "lower_bound": -5, + "upper_bound": 5, + "pad": 0 + }, + "pfcand_dxy": { + "median": 0.0, + "norm_factor": 300, + "replace_inf_value": 0, + "lower_bound": -5, + "upper_bound": 5, + "pad": 0 + }, + "pfcand_dxysig": { + "median": 0, + "norm_factor": 1.0, + "replace_inf_value": 0, + "lower_bound": -5, + "upper_bound": 5, + "pad": 0 + }, + "pfcand_btagEtaRel": { + "median": 1.5, + "norm_factor": 0.5, + "replace_inf_value": 0, + "lower_bound": -5, + "upper_bound": 5, + "pad": 0 + }, + "pfcand_btagPtRatio": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -5, + "upper_bound": 5, + "pad": 0 + }, + "pfcand_btagPParRatio": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -5, + "upper_bound": 5, + "pad": 0 + }, + "pfcand_btagSip3dVal": { + "median": 0, + "norm_factor": 100, + "replace_inf_value": 0, + "lower_bound": -5, + "upper_bound": 5, + "pad": 0 + }, + "pfcand_btagSip3dSig": { + "median": 0, + "norm_factor": 0.5, + "replace_inf_value": 0, + "lower_bound": -5, + "upper_bound": 5, + "pad": 0 + }, + "pfcand_btagJetDistVal": { + "median": 0, + "norm_factor": 40, + "replace_inf_value": 0, + "lower_bound": -5, + "upper_bound": 5, + "pad": 0 + } + }, + "min_length": 16, + "max_length": 100 + }, + "pf_mask": { + "var_names": [ + "pfcand_mask" + ], + "var_infos": { + "pfcand_mask": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + } + }, + "min_length": 16, + "max_length": 100 + }, + "sv_points": { + "var_names": [ + "sv_etarel", + "sv_phirel" + ], + "var_infos": { + "sv_etarel": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "sv_phirel": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + } + }, + "min_length": 1, + "max_length": 10 + }, + "sv_features": { + "var_names": [ + "sv_pt_log", + "sv_mass", + "sv_etarel", + "sv_phirel", + "sv_abseta", + "sv_ntracks", + "sv_normchi2", + "sv_dxy", + "sv_dxysig", + "sv_d3d", + "sv_d3dsig" + ], + "var_infos": { + "sv_pt_log": { + "median": 4, + "norm_factor": 0.6, + "replace_inf_value": 0, + "lower_bound": -5, + "upper_bound": 5, + "pad": 0 + }, + "sv_mass": { + "median": 1.2, + "norm_factor": 0.3, + "replace_inf_value": 0, + "lower_bound": -5, + "upper_bound": 5, + "pad": 0 + }, + "sv_etarel": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "sv_phirel": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "sv_abseta": { + "median": 0.5, + "norm_factor": 1.6, + "replace_inf_value": 0, + "lower_bound": -5, + "upper_bound": 5, + "pad": 0 + }, + "sv_ntracks": { + "median": 3, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -5, + "upper_bound": 5, + "pad": 0 + }, + "sv_normchi2": { + "median": 0.8, + "norm_factor": 0.6, + "replace_inf_value": 0, + "lower_bound": -5, + "upper_bound": 5, + "pad": 0 + }, + "sv_dxy": { + "median": 0.4, + "norm_factor": 0.25, + "replace_inf_value": 0, + "lower_bound": -5, + "upper_bound": 5, + "pad": 0 + }, + "sv_dxysig": { + "median": 7, + "norm_factor": 0.02, + "replace_inf_value": 0, + "lower_bound": -5, + "upper_bound": 5, + "pad": 0 + }, + "sv_d3d": { + "median": 0.5, + "norm_factor": 0.2, + "replace_inf_value": 0, + "lower_bound": -5, + "upper_bound": 5, + "pad": 0 + }, + "sv_d3dsig": { + "median": 7, + "norm_factor": 0.02, + "replace_inf_value": 0, + "lower_bound": -5, + "upper_bound": 5, + "pad": 0 + } + }, + "min_length": 1, + "max_length": 10 + }, + "sv_mask": { + "var_names": [ + "sv_mask" + ], + "var_infos": { + "sv_mask": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + } + }, + "min_length": 1, + "max_length": 10 + } +}