Skip to content

Files

Latest commit

0bc0220 · Dec 12, 2023

History

History

benchmarks

Running the benchmarks

Create a conda environment and activate it:

conda env create -f environment.yml
conda activate lleaves

Modify setup.py to not exclude the benchmark package:

packages=find_packages(exclude=[])  # Used to be: exclude=["benchmark"]

Install the packages in the environment (optionally, in development mode with -e):

python -m pip install --no-build-isolation -e .

Generate the test data and train the necessary models:

./benchmarks/data/setup_data.sh
cd benchmarks
python train_NYC_model.py

Finally, run the benchmarks from within the benchmarks folder:

python benchmark.py