Rustrees

Decision trees, random and causal forests in Rust.

Work in progress, stay tuned!

Using Python library

First, create a virtualenv (this just needs to be done once):

python -m venv .env

Then, activate the virtualenv (needs to be done every time):

source .env/bin/activate

Now, install the requirements (just needs to be done once):

pip install -r requirements.txt

Finally, install the Python library at the local virtual environment with the following command (needs to be done every time you change the Rust code):

maturin develop --release

Now, you can import the library rustrees in Python. This can be done also from Jupyter notebooks. To do so, run the following command:

jupyter notebook

And then import the library in the notebook:

from rustrees.rustrees import DecisionTree, RandomForest
import rustrees.tree as rt