diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..ae4f353 --- /dev/null +++ b/.gitignore @@ -0,0 +1,15 @@ +*.sublime-project +*.sublime-workspace +*.pyc +*.log +*.DS_Store +*.pkl +*.egg-info +*.pdf +*.png +__pycache__ +dist +build +tmp +data + diff --git a/README.md b/README.md index e69de29..387fc20 100644 --- a/README.md +++ b/README.md @@ -0,0 +1,18 @@ +# Data for the CMSSW [RecoHGCal/TICL](https://github.com/cms-sw/cmssw/tree/master/RecoHGCal/TICL) package + +### Quicklinks + +- TICL: + - [Reconstruction example](http://hgcal.web.cern.ch/hgcal/Reconstruction/TICL/) +- TensorFlow: + - [C++ docs](https://www.tensorflow.org/api_docs/cc) + - [CMSSW interface](https://gitlab.cern.ch/mrieger/CMSSW-DNN) + - [PhysicsTools/TensorFlow](https://github.com/cms-sw/cmssw/tree/master/PhysicsTools/TensorFlow) + +### Models + +- `tf_models/energy_id_v*.pb`: TensorFlow model for trackster energy regression and particle ID. + - `v0`: Simple CNN-based approach. The neutral pion, neutral hadron, ambiguous and unknown probabilities are set to a constant value of 0. See the [talk at the Reco/AT meeting](https://indico.cern.ch/event/841640/contributions/3534140/attachments/1896780/3129591/2019-08-23_rieger_hgcal_ticl_eid.pdf) for more info. Input and output tensors: + - `"input"`: Input tensor with dimension `batch x 50 (layers) x 10 (clusters) x 3 (features)`. + - `"output/id_probabilities"`: Output tensor with dimension `batch x 8` representing particle ID "probabilities" (from a softmax output). The probabiltities refer to photon, electron, muon, neutral pion, charged hadron, neutral hadron, ambiguous and unknown cases (in that order). + - `"output/regressed_energy"`: Output tensor with dimension `batch x 1` representing the regressed energy value for the trackster. diff --git a/tf_models/energy_id_v0.pb b/tf_models/energy_id_v0.pb new file mode 100644 index 0000000..cda68e5 Binary files /dev/null and b/tf_models/energy_id_v0.pb differ