Commit 1fb2d6a 1 parent 8fc3f49 commit 1fb2d6a Copy full SHA for 1fb2d6a
File tree 2 files changed +30
-1
lines changed
2 files changed +30
-1
lines changed Original file line number Diff line number Diff line change
1
+ ## Running the benchmarks
2
+
3
+ Create a conda environment and activate it:
4
+ ``` commandline
5
+ conda env create -f environment.yml
6
+ conda activate lleaves
7
+ ```
8
+
9
+ Modify [ setup.py] ( ../setup.py ) to not exclude the ` benchmark ` package:
10
+ ``` python
11
+ packages= find_packages(exclude = []) # Used to be: exclude=["benchmark"]
12
+ ```
13
+
14
+ Install the packages in the environment (optionally, in development mode with ` -e ` ):
15
+ ``` commandline
16
+ python -m pip install --no-build-isolation -e .
17
+ ```
18
+
19
+ Generate the test data and train the necessary models:
20
+ ``` commandline
21
+ benchmarks/data/setup_data.sh
22
+ python benchmarks/train_NYC_model.py
23
+ ```
24
+
25
+ Finally, run the benchmarks from within the benchmarks folder:
26
+ ``` commandline
27
+ cd benchmarks
28
+ python benchmark.py
29
+ ```
Original file line number Diff line number Diff line change @@ -10,7 +10,7 @@ dependencies:
10
10
# testing
11
11
- pytest
12
12
- hypothesis
13
- - lightgbm
13
+ - lightgbm<4 # Pin needed until https://github.com/onnx/onnxmltools/issues/645 is fixed
14
14
- pre-commit
15
15
- pyarrow
16
16
- pandas
You can’t perform that action at this time.
0 commit comments