Metadata-Version: 2.4
Name: directrs
Version: 0.1.1
Summary: DirectRS: extract geometry from decision trees
License-File: LICENSE
Requires-Python: >=3.10
Requires-Dist: joblib>=1.3
Requires-Dist: matplotlib>=3.7
Requires-Dist: numpy>=1.24
Requires-Dist: pandas>=2.0
Requires-Dist: rich>=13.0
Requires-Dist: scikit-learn>=1.3
Requires-Dist: seaborn>=0.12
Requires-Dist: typer>=0.9
Description-Content-Type: text/markdown

# directrs

**DirectRS** — extract geometry from decision trees to build piecewise-linear models.

Based on Agus Sudjianto's paper *"Linear Models Hiding Inside Your Decision Tree"*.

## Installation

```bash
pixi install
```

## Usage

```bash
# Classification
pixi run cli train --task classification --input data.csv --target target --output-dir output/

# Regression
pixi run cli train --task regression --input data.csv --target target --output-dir output/

# Predict
pixi run cli predict --model model.pkl --input new_data.csv --output predictions.csv
```

## Run tests

```bash
pixi run test
```
