README.md 765 Bytes
Newer Older
Niklas Uwe Langner's avatar
Niklas Uwe Langner committed
1
2
3
4
5
6
# Keras EdgeConv-Layer

General implementation of the EdgeConv-Block as described in [ParticleNet: Jet Tagging via Particle Clouds](https://arxiv.org/abs/1902.08570).

## Using the layer

7
In order to use the layer class found in [edgeconv.py](edgeconv.py), a h-function needs to be defined. This function should take a list of two Keras tensors of length C with C being the dimension of the features. The h-function might include other Keras layers. Setting the number of k nearest neighbors to be considered using the `next_neighbors` argument, the EdgeConv layer can be implemented as demonstrated in [test.py](test.py).
Niklas Uwe Langner's avatar
Niklas Uwe Langner committed
8
9
10

## Acknowledgement
This implementation borrows code from the ParticleNet tensorflow [implementation](https://github.com/hqucms/ParticleNet).