Skip to content

Commit 8d14c4a

Browse files
authored
Merge pull request #3 from fastmachinelearning/hls4ml_jet
hls4ml reference 3 layer MLP for jet tagging
2 parents f2dffdc + de731b9 commit 8d14c4a

File tree

2 files changed

+21
-0
lines changed

2 files changed

+21
-0
lines changed
Binary file not shown.

models/JetTagging/README.md

Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,21 @@
1+
# Jet Tagging
2+
3+
The jet tagging dataset is an open dataset of simulated jets from collisions at the Large Hadron Collider. It's described in more detail in the [first hls4ml paper](https://iopscience.iop.org/article/10.1088/1748-0221/13/07/P07027), and available online at [Zenodo](https://zenodo.org/record/3602254#.YiePu_XMLUI), and [OpenML](https://www.openml.org/d/42468).
4+
5+
To access the dataset quickly:
6+
```
7+
from sklearn.datasets import fetch_openml
8+
data = fetch_openml('hls4ml_lhc_jets_hlf')
9+
```
10+
11+
## Models
12+
13+
The model `qkeras_jettagging.onnx` is trained in part 4 of the [hls4ml tutorial](https://github.com/fastmachinelearning/hls4ml-tutorial) - including some recommended dataset preprocessing.
14+
It's an MLP with 3 hidden layers, with ReLU activations and a Softmax layer at the output.
15+
16+
The models here were trained originally using `QKeras`, then translated to `QONNX` using the in-development converter `qonnx.converters.from_keras`.
17+
The original `QKeras` model is also available from the `hls4ml` [example models](https://github.com/fastmachinelearning/example-models/tree/0d4cc7277eac9bb9020e3d73a992dc15dbdcce4e/keras) (`qkeras_3layer` model).
18+
19+
| weight bits | bias bits | activation bits | accuracy |
20+
| --- | --- | --- | --- |
21+
| 6 | 6 | 6 | 76.2 % |

0 commit comments

Comments
 (0)