Skip to main content

Optimal Model Trees using OR-Tools

Project description

OptimalModelTree (OMT)

A mixed-integer optimization-based decision tree framework for classification and regression.

This package implements Optimal Model Trees (OMT), where the tree structure and prediction rules are learned jointly via a mathematical optimization model (MILP) using Gurobi.


🚀 Features

  • Optimal decision tree construction via mixed-integer programming
  • Supports:
    • Classification (binary & multiclass)
    • Regression
  • Parallel and oblique splits
  • Regularized models (L1-style sparsity control)
  • Warm-start support for faster optimization
  • Scikit-learn compatible API (fit, predict)

📦 Installation

From PyPI

pip install optimal-omt

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

optimal_omt-0.1.3.tar.gz (8.8 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

optimal_omt-0.1.3-py3-none-any.whl (9.3 kB view details)

Uploaded Python 3

File details

Details for the file optimal_omt-0.1.3.tar.gz.

File metadata

  • Download URL: optimal_omt-0.1.3.tar.gz
  • Upload date:
  • Size: 8.8 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.12

File hashes

Hashes for optimal_omt-0.1.3.tar.gz
Algorithm Hash digest
SHA256 4e005667476e25b74aede8627817435f4e376eec55b9b4bb8ee3e1a8924a8bc9
MD5 eedf2146fc13c94dc997c36802235b6a
BLAKE2b-256 f165d70c23a8f292847451de49dac90db7a1f3bf8377a3023ddd85d159d6ba40

See more details on using hashes here.

Provenance

The following attestation bundles were made for optimal_omt-0.1.3.tar.gz:

Publisher: release.yml on sabinoroselli/OptimalModelTree

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

File details

Details for the file optimal_omt-0.1.3-py3-none-any.whl.

File metadata

  • Download URL: optimal_omt-0.1.3-py3-none-any.whl
  • Upload date:
  • Size: 9.3 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.12

File hashes

Hashes for optimal_omt-0.1.3-py3-none-any.whl
Algorithm Hash digest
SHA256 9d226466f426c94191e876ed6fe2b1995e09276ac8fa64f5b4b260d6e343d8ae
MD5 4a0daf86f2bfda45bf26c2bf7c7acb64
BLAKE2b-256 928f26e7b6e68bb68d7a4677221a349900234a7965470ea34908b31390d6f94c

See more details on using hashes here.

Provenance

The following attestation bundles were made for optimal_omt-0.1.3-py3-none-any.whl:

Publisher: release.yml on sabinoroselli/OptimalModelTree

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page