diff --git a/PNetMuonId/model.onnx b/PNetMuonId/model.onnx new file mode 100644 index 0000000..ff53f64 Binary files /dev/null and b/PNetMuonId/model.onnx differ diff --git a/PNetMuonId/preprocess.json b/PNetMuonId/preprocess.json new file mode 100644 index 0000000..5fa0206 --- /dev/null +++ b/PNetMuonId/preprocess.json @@ -0,0 +1,459 @@ +{ + "output_names": [ + "label_lightfake", + "label_prompt", + "label_tau", + "label_heavy" + ], + "input_names": [ + "pf_points", + "pf_features", + "pf_mask", + "sv_points", + "sv_features", + "sv_mask", + "high_level" + ], + "pf_points": { + "var_names": [ + "PF_phi_rel", + "PF_eta_rel" + ], + "var_infos": { + "PF_phi_rel": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "PF_eta_rel": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + } + }, + "min_length": 16, + "max_length": 60 + }, + "pf_features": { + "var_names": [ + "PF_pt", + "PF_eta_rel", + "PF_phi_rel", + "PF_charge", + "PF_isElectron", + "PF_isMuon", + "PF_isNeutralHadron", + "PF_isPhoton", + "PF_isChargedHadron", + "PF_fromPV", + "PF_puppiWeightNoLep", + "PF_hcalFraction", + "PF_dzSig", + "PF_dxySig" + ], + "var_infos": { + "PF_pt": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "PF_eta_rel": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "PF_phi_rel": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "PF_charge": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "PF_isElectron": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "PF_isMuon": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "PF_isNeutralHadron": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "PF_isPhoton": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "PF_isChargedHadron": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "PF_fromPV": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "PF_puppiWeightNoLep": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "PF_hcalFraction": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "PF_dzSig": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "PF_dxySig": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + } + }, + "min_length": 16, + "max_length": 60 + }, + "pf_mask": { + "var_names": [ + "PF_mask" + ], + "var_infos": { + "PF_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": 60 + }, + "sv_points": { + "var_names": [ + "SV_eta", + "SV_phi" + ], + "var_infos": { + "SV_eta": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "SV_phi": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + } + }, + "min_length": 1, + "max_length": 5 + }, + "sv_features": { + "var_names": [ + "SV_pt", + "SV_mass", + "SV_phi", + "SV_eta", + "SV_dlenSig", + "SV_dxy", + "SV_ndof", + "SV_chi2", + "SV_cospAngle", + "SV_nTracks" + ], + "var_infos": { + "SV_pt": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "SV_mass": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "SV_phi": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "SV_eta": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "SV_dlenSig": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "SV_dxy": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "SV_ndof": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "SV_chi2": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "SV_cospAngle": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "SV_nTracks": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + } + }, + "min_length": 1, + "max_length": 5 + }, + "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": 5 + }, + "high_level": { + "var_names": [ + "MuonSelected_dxy", + "MuonSelected_dz", + "MuonSelected_sip3d", + "MuonSelected_segmentComp", + "MuonSelected_LepGood_jetDF", + "MuonSelected_LepGood_jetPtRatio", + "MuonSelected_LepGood_jetNDauChargedMVASel", + "MuonSelected_LepGood_dz", + "MuonSelected_LepGood_eta", + "MuonSelected_LepGood_miniRelIsoNeutral", + "MuonSelected_LepGood_jetPtRelv2", + "MuonSelected_LepGood_miniRelIsoCharged", + "MuonSelected_mvaId" + ], + "var_infos": { + "MuonSelected_dxy": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "MuonSelected_dz": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "MuonSelected_sip3d": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "MuonSelected_segmentComp": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "MuonSelected_LepGood_jetDF": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "MuonSelected_LepGood_jetPtRatio": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "MuonSelected_LepGood_jetNDauChargedMVASel": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "MuonSelected_LepGood_dz": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "MuonSelected_LepGood_eta": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "MuonSelected_LepGood_miniRelIsoNeutral": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "MuonSelected_LepGood_jetPtRelv2": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "MuonSelected_LepGood_miniRelIsoCharged": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + }, + "MuonSelected_mvaId": { + "median": 0, + "norm_factor": 1, + "replace_inf_value": 0, + "lower_bound": -1e+32, + "upper_bound": 1e+32, + "pad": 0 + } + }, + "var_length": 1 + } +} \ No newline at end of file