ipytreecraft 0.1.0__py3-none-any.whl

ipytreecraft-0.1.0.data/data/share/jupyter/lab/settings/overrides.d/treecraft.json

@@ -0,0 +1,10 @@
+{
+  "@jupyterlab/notebook-extension:tracker": {
+    "windowingMode": "none"
+  },
+  "@jupyter-widgets/jupyterlab-manager:plugin": {
+    "saveState": true
+  }
+}
+
+

ipytreecraft-0.1.0.dist-info/METADATA

@@ -0,0 +1,85 @@
+Metadata-Version: 2.4
+Name: ipytreecraft
+Version: 0.1.0
+Summary: Incremental, interactive decision tree builder for risk specialists
+Author-email: Mark <mark@example.com>
+Classifier: Framework :: Jupyter
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Requires-Python: >=3.11
+Requires-Dist: joblib>=1.3
+Requires-Dist: numpy>=1.26
+Requires-Dist: pandas>=2.0
+Requires-Dist: pydantic>=2.0
+Requires-Dist: scikit-learn<2.0,>=1.5
+Provides-Extra: api
+Requires-Dist: fastapi>=0.110; extra == 'api'
+Requires-Dist: python-multipart>=0.0.9; extra == 'api'
+Requires-Dist: slowapi>=0.1.9; extra == 'api'
+Requires-Dist: uvicorn[standard]>=0.29; extra == 'api'
+Provides-Extra: dev
+Requires-Dist: httpx>=0.27; extra == 'dev'
+Requires-Dist: pytest-cov>=5.0; extra == 'dev'
+Requires-Dist: pytest>=8.0; extra == 'dev'
+Requires-Dist: ruff>=0.4; extra == 'dev'
+Provides-Extra: jupyter
+Requires-Dist: anywidget>=0.9.13; extra == 'jupyter'
+Requires-Dist: jupyterlab>=4.0; extra == 'jupyter'
+Description-Content-Type: text/markdown
+
+# Treecraft
+
+[![CI](https://github.com/markdedeuge/treecraft/actions/workflows/ci.yml/badge.svg)](https://github.com/markdedeuge/treecraft/actions/workflows/ci.yml) [![Coverage](https://img.shields.io/badge/coverage-96%25-success)](https://github.com/markdedeuge/treecraft) [![PyPI version](https://img.shields.io/pypi/v/treecraft.svg)](https://pypi.org/project/treecraft/) [![Downloads](https://img.shields.io/pypi/dm/treecraft.svg)](https://pypi.org/project/treecraft/) [![License](https://img.shields.io/badge/license-MIT-success)](https://github.com/markdedeuge/treecraft/blob/main/LICENSE)
+
+Treecraft is an interactive decision tree framework. Treecraft allows analysts to manually evaluate, override, and prune decision boundaries utilizing an interactive UI interface, while automatically refitting subtrees optimally beneath manual split points.
+
+![Treecraft Interface](docs/treecraftui.png)
+
+The framework extends standard Scikit-Learn. Modifications made visually in the interface alter splits in the underlying tree graph. The resulting model can then be exported as a standard, fitted `DecisionTreeClassifier` or `DecisionTreeRegressor` compatible with standard sklearn pipelines.
+
+## Installation
+
+This module provides an interactive interface built upon `anywidget` for integration into Jupyter ecosystems.
+
+Install from source locally via standard python package deployment:
+
+```bash
+pip install -e .
+```
+
+## Usage
+
+Initiate the widget by supplying a `pandas.DataFrame` and declaring the target feature.
+
+```python
+import pandas as pd
+from sklearn.datasets import load_iris
+from treecraft.widget import TreecraftWidget
+
+data = load_iris()
+df = pd.DataFrame(data.data, columns=data.feature_names)
+df['target'] = data.target
+
+# Instantiate the interactive widget
+widget = TreecraftWidget(df=df, target='target', model_id='iris_model')
+widget
+```
+
+Once the interactive visualization opens, users can apply initial hyperparameters, manually navigate tree partitions, append/remove split points, and verify node performance across configurable train/validation splits.
+
+### Pipeline Extraction
+
+After modifying the boundaries visually to conform to the required constraints, extract the native Scikit-Learn estimator.
+
+```python
+from sklearn.pipeline import make_pipeline
+
+# Extract the fitted Scikit-Learn estimator structure
+clf = widget.export_sklearn()
+pipeline = make_pipeline(clf)
+
+# Execute predictions
+X_test = df.drop(columns=['target'])
+predictions = pipeline.predict(X_test)
+```
+

ipytreecraft-0.1.0.dist-info/RECORD

@@ -0,0 +1,21 @@
+treecraft/__init__.py,sha256=4OzJlLicc8p4dFR5iXtQZ_0A0r6LAABbEJz9yBcdlhw,530
+treecraft/widget.py,sha256=_Ea3bhUmvNsGg39sbSwA8iA77g91ejBDHrBGqv9ae_c,20610
+treecraft/engine/__init__.py,sha256=Bi742cK5TaARof69Mct6pTLrLWMf6cyNEeNOAOh-ZWc,49
+treecraft/engine/constraint_manager.py,sha256=23qQsGqTJLLuj7iHhA8I_F4ip5QF_DxulPZdZ3pvjrs,2108
+treecraft/engine/incremental_tree.py,sha256=Aw49INnQtvlYXAs6qijlfva4oF4P9SPibi6i4HZWevo,51143
+treecraft/engine/metrics.py,sha256=3Bmh1pYtnvZLK95_1x5_eChOpvNalj4UFuJz0c2yet0,5281
+treecraft/engine/subtree_fitter.py,sha256=4_X6ucSbyo-xOv5ZWjLYrap_6bRhNNpb4qWlwa0LYSo,2691
+treecraft/engine/tree_state.py,sha256=ggjloDrydLoxIizVomlMWngHbfkezq2OTk0xWi6r0Kg,16318
+treecraft/engine/tree_stitcher.py,sha256=NVlhFqwSRBeO4SqZEgqjPWT_spphr3DoDcQ-p-bEVco,11000
+treecraft/models/__init__.py,sha256=OG-5u75I0WMtqIXLNi-zI7xNPggiv6msefUmGVEGz3E,47
+treecraft/models/enums.py,sha256=9ne2S2lot7CB5E39pRiygOZwgSjRQPCO1NtL0NYNQco,1480
+treecraft/models/parameter_introspector.py,sha256=Pv2JQmESldpo-gCnPhkRT79aTXXZ02rSwZKP4uX9yTA,12817
+treecraft/models/schemas.py,sha256=gZD8luL6IYs3eevfp_liv6yl0KutESiS94RlSHmww7U,15005
+treecraft/utils/__init__.py,sha256=dOYPix5mbfd6guG-bxYgZTiDkSEQQyTBIJjlejehVs8,39
+treecraft/utils/export.py,sha256=Q340B3qJ_z--8YIjN8MfHw2bHBhiGZ6cX6jXNvz7RKs,6148
+treecraft/utils/type_inference.py,sha256=U4AehDQzMa5sEa5nID-fEMyufSXX3tauwycBLLWuXMU,3009
+treecraft/static/widget.js,sha256=KyNSFMXwinTt8pTAe8sNUTdFgMOQzzVPz-if6a36gLE,4472839
+ipytreecraft-0.1.0.data/data/share/jupyter/lab/settings/overrides.d/treecraft.json,sha256=POfn1Hhafiec2hUk4LEZZvDFPZDpYDu3p14fVZTcyhE,161
+ipytreecraft-0.1.0.dist-info/METADATA,sha256=UW5OeS3_Mk0wWB0iFgjrGUy9kLTMH4QiyzsxV_Yn9cQ,3561
+ipytreecraft-0.1.0.dist-info/WHEEL,sha256=QccIxa26bgl1E6uMy58deGWi-0aeIkkangHcxk2kWfw,87
+ipytreecraft-0.1.0.dist-info/RECORD,,

ipytreecraft-0.1.0.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.29.0
+Root-Is-Purelib: true
+Tag: py3-none-any

treecraft/__init__.py

@@ -0,0 +1,23 @@
+"""Treecraft: Incremental, interactive decision tree builder."""
+
+from treecraft.engine.incremental_tree import IncrementalTreeEngine
+from treecraft.models.schemas import (
+    CrossValidationParams,
+    DatasetMeta,
+    SplitOverride,
+    TrainTestSplitParams,
+    TreeConfig,
+)
+from treecraft.widget import TreecraftWidget
+
+__version__ = "0.1.0"
+
+__all__ = [
+    "IncrementalTreeEngine",
+    "TreecraftWidget",
+    "DatasetMeta",
+    "TreeConfig",
+    "TrainTestSplitParams",
+    "CrossValidationParams",
+    "SplitOverride",
+]

treecraft/engine/__init__.py

@@ -0,0 +1 @@
+"""Core engine for incremental tree building."""

treecraft/engine/constraint_manager.py

@@ -0,0 +1,58 @@
+"""ConstraintManager — tracks pinned nodes and manual split overrides."""
+
+from __future__ import annotations
+
+from treecraft.models.schemas import SplitOverride
+
+
+class ConstraintManager:
+    """Manages which tree nodes have user-pinned split overrides.
+
+    A pinned node's split (feature + threshold/categories) is frozen
+    during incremental re-training — the algorithm only optimises the
+    subtrees below pinned nodes.
+    """
+
+    def __init__(self) -> None:
+        self._pinned: dict[int, SplitOverride] = {}
+
+    def pin_node(self, node_id: int, override: SplitOverride) -> None:
+        """Pin *node_id* with the given split override."""
+        self._pinned[node_id] = override
+
+    def unpin_node(self, node_id: int) -> None:
+        """Remove the pin from *node_id* (no-op if not pinned)."""
+        self._pinned.pop(node_id, None)
+
+    def get_pinned_nodes(self) -> dict[int, SplitOverride]:
+        """Return a copy of all pinned node overrides."""
+        return dict(self._pinned)
+
+    def is_frozen(self, node_id: int) -> bool:
+        """Return ``True`` if *node_id* is pinned."""
+        return node_id in self._pinned
+
+    def clear(self) -> None:
+        """Remove all pins."""
+        self._pinned.clear()
+
+    def remap_ids(self, old_to_new: dict[int, int]) -> None:
+        """Remap pinned node IDs after a stitch operation.
+
+        Pins whose old ID does not appear in *old_to_new* are
+        discarded (the node was removed during the stitch).
+        """
+        remapped: dict[int, SplitOverride] = {}
+        for old_id, override in self._pinned.items():
+            new_id = old_to_new.get(old_id)
+            if new_id is not None:
+                # Update the node_id inside the override as well.
+                override = override.model_copy(update={"node_id": new_id})
+                remapped[new_id] = override
+        self._pinned = remapped
+
+    def copy(self) -> ConstraintManager:
+        """Return a deep copy of this manager."""
+        cm = ConstraintManager()
+        cm._pinned = {k: v.model_copy() for k, v in self._pinned.items()}
+        return cm