Skip to content

Commit

Permalink
Merge pull request #1 from riga/master
Browse files Browse the repository at this point in the history
Initial commit with CNN-based energy regression and ID tf model.
  • Loading branch information
cmsbuild authored Sep 25, 2019
2 parents 748e991 + 197566b commit 1e9fc29
Show file tree
Hide file tree
Showing 3 changed files with 33 additions and 0 deletions.
15 changes: 15 additions & 0 deletions .gitignore
Original file line number Diff line number Diff line change
@@ -0,0 +1,15 @@
*.sublime-project
*.sublime-workspace
*.pyc
*.log
*.DS_Store
*.pkl
*.egg-info
*.pdf
*.png
__pycache__
dist
build
tmp
data

18 changes: 18 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
@@ -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.
Binary file added tf_models/energy_id_v0.pb
Binary file not shown.

0 comments on commit 1e9fc29

Please sign in to comment.