Skip to main content

TensorFlow implementation of the Lorentz Boost Network (LBN).

Project description

Lorentz Boost Network (LBN) pipeline status

TensorFlow implementation of the Lorentz Boost Network from arXiv:1812.09722 [hep-ex].

Original repository:

Usage example

from lbn import LBN

# initialize the LBN, set 10 combinations and pairwise boosting
lbn = LBN(10, boost_mode=LBN.PAIRS)

# create a feature tensor based on input four-vectors
features = lbn(four_vectors)

# use the features as input for a subsequent, application-specific network

Installation and dependencies

Via pip:

pip install lbn

NumPy and TensorFlow are the only dependencies.


Tests should be run for Python 2 and 3. The following commands assume you are root directory of the LBN respository:

python -m unittest test

# or via docker, python 2
docker run --rm -v `pwd`:/root/lbn -w /root/lbn tensorflow/tensorflow:latest python -m unittest test

# or via docker, python 3
docker run --rm -v `pwd`:/root/lbn -w /root/lbn tensorflow/tensorflow:latest-py3 python -m unittest test


If you like to contribute, we are happy to receive pull requests. Just make sure to add new test cases and run the tests. Also, please use a coding style that is compatible with our .flake8 config.


Project details

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

lbn-1.0.1.tar.gz (12.0 kB view hashes)

Uploaded Source

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page