Published March 25, 2024 | Version 0.1
Software Open

D4.1 Framework for Learning with Robustness Guarantees

Creators

  • 1. ROR icon KU Leuven

Description

This Python software package takes a pre-trained tree ensemble model and compresses it in a top-down fashion. Starting from the root and going down level per level, it prunes away subtrees by fitting coefficients.

commit: 55a8cb1

Files

tree_compress-55a8cb13246cdb57194123e80245fa77a42315f0.zip

Files (327.4 kB)

Additional details

Funding

TUPLES – Trustworthy Planning and Scheduling with Learning and Explanations 101070149
European Commission

Dates

Available
2024-03

Software

Repository URL
https://github.com/laudv/tree_compress
Programming language
Python
Development Status
Active