/coppice

Python decision tree and ensemble package using a rust backend.

Primary LanguageRustOtherNOASSERTION

coppice

A simple decision tree and random forest library using a Rust backend.

Installation

pip install coppice

Usage

Python

import numpy as np

from coppice import _coppice

x = ... # [w, n]
y = ... # [n]

x = x.astype(np.float32)
y = y.astype(np.uint32)

_coppice.train_tree_np(x, y, max_depth=5)

_coppice.train_forest_np(x, y, max_depth=5, num_trees=10)

CLI

usage: coppice_ [-h] [--model-type {tree,forest}] [--num-trees NUM_TREES]
                [--max-depth MAX_DEPTH]
                x-path y-path

positional arguments:
  x-path
  y-path

options:
  -h, --help            show this help message and exit
  --model-type {tree,forest}
  --num-trees NUM_TREES
  --max-depth MAX_DEPTH

Development

git clone git@github.com:JBwdn/coppice.git
cd coppice
pip install -e ".[dev]"
maturin develop
pre-commit install