Installation

OMLT requires Python >= 3.6. The most stable OMLT version can be installed using the PyPI package index. This will also install the required depencies. Simply run:

pip install omlt

If using the latest un-released version, install from the github repository and install locally

git clone https://github.com/cog-imperial/OMLT.git
cd OMLT
pip install .

Optional Requirements

OMLT can import sequential Keras models which requires a working installation of tensorflow:

pip install tensorflow

OMLT can also import neural network and gradient boosted tree models using ONNX. This requires installing the ONNX interface:

pip install onnx