tree-praxis 0.0.13__tar.gz

tree_praxis-0.0.13/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 Zakk Heile
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

tree_praxis-0.0.13/MANIFEST.in

@@ -0,0 +1 @@
+recursive-include src/praxis/cpp *.cpp *.hpp *.h

tree_praxis-0.0.13/PKG-INFO

@@ -0,0 +1,27 @@
+Metadata-Version: 2.4
+Name: tree-praxis
+Version: 0.0.13
+Summary: Rashomon set tools via PRAXIS, including approximation, exact calculation, and variable importance utilities.
+Author: Zakk Heile, Hayden McTavish, Varun Babbar, Margo Seltzer, Cynthia Rudin
+License: MIT
+Project-URL: Homepage, https://github.com/zakk-h/LicketyRESPLIT
+Project-URL: Repository, https://github.com/zakk-h/LicketyRESPLIT
+Project-URL: Issues, https://github.com/zakk-h/LicketyRESPLIT/issues
+Keywords: interpretable machine learning,decision trees,rashomon set,optimization,variable importance
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Science/Research
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: C++
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: numpy>=1.23
+Requires-Dist: matplotlib>=3.6
+Requires-Dist: scikit-learn>=1.1
+Requires-Dist: pandas>=1.5
+Dynamic: license-file

tree_praxis-0.0.13/pyproject.toml

@@ -0,0 +1,50 @@
+[build-system]
+requires = ["setuptools>=64", "wheel", "pybind11>=2.13", "numpy>=1.23"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "tree-praxis"
+version = "0.0.13"
+description = "Rashomon set tools via PRAXIS, including approximation, exact calculation, and variable importance utilities."
+readme = "README.md"
+requires-python = ">=3.9"
+license = {text = "MIT"}
+authors = [
+  { name = "Zakk Heile" },
+  { name = "Hayden McTavish" },
+  { name = "Varun Babbar" },
+  { name = "Margo Seltzer" },
+  { name = "Cynthia Rudin" }
+]
+dependencies = [
+  "numpy>=1.23",
+  "matplotlib>=3.6",
+  "scikit-learn>=1.1",
+  "pandas>=1.5"
+]
+keywords = [
+  "interpretable machine learning",
+  "decision trees",
+  "rashomon set",
+  "optimization",
+  "variable importance"
+]
+classifiers = [
+  "Development Status :: 3 - Alpha",
+  "Intended Audience :: Science/Research",
+  "Programming Language :: Python :: 3",
+  "Programming Language :: Python :: 3.9",
+  "Programming Language :: Python :: 3.10",
+  "Programming Language :: Python :: 3.11",
+  "Programming Language :: Python :: 3.12",
+  "Programming Language :: C++",
+  "Topic :: Scientific/Engineering :: Artificial Intelligence"
+]
+
+[project.urls]
+Homepage = "https://github.com/zakk-h/LicketyRESPLIT"
+Repository = "https://github.com/zakk-h/LicketyRESPLIT"
+Issues = "https://github.com/zakk-h/LicketyRESPLIT/issues"
+
+[tool.setuptools.packages.find]
+where = ["src"]

tree_praxis-0.0.13/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

tree_praxis-0.0.13/setup.py

@@ -0,0 +1,103 @@
+import os
+import platform
+from setuptools import setup, Extension
+from setuptools.command.build_ext import build_ext
+import pybind11
+
+
+def is_truthy_env(name: str) -> bool:
+    return os.environ.get(name, "").lower() in ("1", "true", "yes", "on")
+
+
+class BuildExt(build_ext):
+    c_opts = {
+        "msvc": [
+            "/O2",
+            "/std:c++17",
+        ],
+        "unix": [
+            "-O3",
+            "-DNDEBUG",
+            "-funroll-loops",
+        ],
+    }
+
+    l_opts = {
+        "msvc": [],
+        "unix": [],
+    }
+
+    def build_extensions(self):
+        ct = self.compiler.compiler_type
+        system = platform.system().lower() # "linux", "darwin", "windows"
+        machine = platform.machine().lower() # "x86_64", "amd64", "arm64", "aarch64"
+
+        opts = self.c_opts.get(ct, []).copy()
+        link_opts = self.l_opts.get(ct, []).copy()
+
+        if ct == "unix":
+            opts += ["-std=c++17", "-fPIC"]
+
+            if system != "darwin":
+                opts += ["-flto"]
+                link_opts += ["-flto", "-lm"]
+
+            # hardware popcount support.
+            if machine in ("x86_64", "amd64"):
+                # enables x86 POPCNT instruction.
+                opts += ["-mpopcnt"]
+                print("** Building PRAXIS with x86 POPCNT support")
+
+            elif machine in ("arm64", "aarch64"):
+                # Apple Silicon / ARM64 has popcount through ARM/NEON codegen.
+                print("** Building PRAXIS on ARM64; skipping x86 -mpopcnt")
+
+            else:
+                print(f"** Building PRAXIS on unknown Unix arch {machine}; skipping popcount-specific flags")
+
+        elif ct == "msvc":
+            # Windows x64
+            print("** Building PRAXIS with MSVC safe flags")
+
+        aggressive = is_truthy_env("AGGRESSIVE")
+
+        if aggressive and ct == "unix":
+            if machine in ("x86_64", "amd64"):
+                opts += [
+                    "-mbmi",
+                    "-mbmi2",
+                    "-mavx2",
+                ]
+                print("** Building PRAXIS with additional aggressive x86 flags")
+            elif machine in ("arm64", "aarch64"):
+                # potential speedup for local builds
+                # opts += ["-mcpu=native"]
+                print("** AGGRESSIVE requested on ARM64; no extra portable flags added")
+        elif aggressive and ct != "unix":
+            print("** AGGRESSIVE requested on non-Unix compiler; using safe flags")
+
+        for ext in self.extensions:
+            ext.extra_compile_args = opts
+            ext.extra_link_args = link_opts
+
+        build_ext.build_extensions(self)
+
+
+ext_modules = [
+    Extension(
+        "praxis._core",
+        sources=[
+            "src/praxis/_core.cpp",
+        ],
+        include_dirs=[
+            pybind11.get_include(),
+            "src/praxis/cpp",
+        ],
+        language="c++",
+    ),
+]
+
+setup(
+    ext_modules=ext_modules,
+    cmdclass={"build_ext": BuildExt},
+)