endgame-ml 1.0.0__tar.gz

endgame_ml-1.0.0/.github/ISSUE_TEMPLATE/bug_report.yml

@@ -0,0 +1,59 @@
+name: Bug Report
+description: Report a bug in Endgame
+labels: [bug]
+
+body:
+  - type: markdown
+    attributes:
+      value: |
+        Thank you for reporting a bug. Please fill out the sections below so we can reproduce and fix the issue.
+
+  - type: textarea
+    id: description
+    attributes:
+      label: Description
+      description: A clear description of the bug.
+      placeholder: What happened? What did you expect to happen?
+    validations:
+      required: true
+
+  - type: textarea
+    id: reproduction
+    attributes:
+      label: Steps to Reproduce
+      description: Minimal code or steps to reproduce the bug.
+      placeholder: |
+        ```python
+        import endgame as eg
+        # ...
+        ```
+    validations:
+      required: true
+
+  - type: textarea
+    id: expected
+    attributes:
+      label: Expected Behavior
+      description: What you expected to happen instead.
+    validations:
+      required: true
+
+  - type: textarea
+    id: traceback
+    attributes:
+      label: Traceback
+      description: Full error traceback, if applicable.
+      render: python
+
+  - type: textarea
+    id: environment
+    attributes:
+      label: Environment
+      description: Version and platform details.
+      placeholder: |
+        - Endgame version:
+        - Python version:
+        - OS:
+        - Relevant optional dependencies (e.g., xgboost, torch):
+    validations:
+      required: true

endgame_ml-1.0.0/.github/ISSUE_TEMPLATE/feature_request.yml

@@ -0,0 +1,71 @@
+name: Feature Request
+description: Suggest a new feature or improvement
+labels: [enhancement]
+
+body:
+  - type: markdown
+    attributes:
+      value: |
+        Thank you for suggesting an improvement to Endgame. Please describe the feature and why it would be valuable.
+
+  - type: textarea
+    id: description
+    attributes:
+      label: Description
+      description: A clear description of the feature you would like.
+    validations:
+      required: true
+
+  - type: textarea
+    id: motivation
+    attributes:
+      label: Motivation
+      description: Why is this feature needed? What problem does it solve? Include links to papers or implementations if relevant.
+    validations:
+      required: true
+
+  - type: textarea
+    id: proposed-solution
+    attributes:
+      label: Proposed Solution
+      description: How do you think this should work? Include API sketches if you have them.
+      placeholder: |
+        ```python
+        from endgame.models import NewModel
+        model = NewModel(param=value)
+        model.fit(X, y)
+        ```
+
+  - type: dropdown
+    id: module
+    attributes:
+      label: Module
+      description: Which module does this relate to?
+      options:
+        - models
+        - ensemble
+        - calibration
+        - preprocessing
+        - validation
+        - visualization
+        - timeseries
+        - signal
+        - vision
+        - nlp
+        - audio
+        - anomaly
+        - tune
+        - benchmark
+        - quick
+        - automl
+        - mcp
+        - explain
+        - fairness
+        - other
+
+  - type: checkboxes
+    id: contribution
+    attributes:
+      label: Willingness to Contribute
+      options:
+        - label: I would be willing to submit a PR for this feature

endgame_ml-1.0.0/.github/PULL_REQUEST_TEMPLATE.md

@@ -0,0 +1,19 @@
+## Description
+
+<!-- What does this PR do? Why is it needed? Link to any related issues. -->
+
+## Changes
+
+<!-- Bullet-point summary of the changes. -->
+
+-
+
+## Checklist
+
+- [ ] Code follows the sklearn API (`fit`/`predict`/`transform`)
+- [ ] Tests are included and pass (`pytest tests/ -v`)
+- [ ] Linting passes (`ruff check endgame/`)
+- [ ] New public classes are exported from `__init__.py`
+- [ ] Docstrings are present (Google style, with parameters and examples)
+- [ ] Optional dependencies are handled with graceful import errors
+- [ ] `CHANGELOG.md` updated (if applicable)

endgame_ml-1.0.0/.github/workflows/ci.yml

@@ -0,0 +1,86 @@
+name: CI
+
+on:
+  push:
+    branches: [main]
+  pull_request:
+    branches: [main]
+
+concurrency:
+  group: ${{ github.workflow }}-${{ github.ref }}
+  cancel-in-progress: true
+
+jobs:
+  lint:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v4
+
+      - uses: actions/setup-python@v5
+        with:
+          python-version: "3.12"
+
+      - name: Install dependencies
+        run: pip install ruff
+
+      - name: Lint
+        run: ruff check endgame/
+
+  test:
+    runs-on: ubuntu-latest
+    strategy:
+      fail-fast: false
+      matrix:
+        python-version: ["3.10", "3.11"]
+    steps:
+      - uses: actions/checkout@v4
+
+      - uses: actions/setup-python@v5
+        with:
+          python-version: ${{ matrix.python-version }}
+          cache: pip
+
+      - name: Install dependencies
+        run: pip install -e ".[dev,tabular,calibration,explain,fairness,benchmark,anomaly,tracking,mcp]"
+
+      - name: Run tests
+        run: >
+          pytest tests/ -v --tb=short
+          --cov=endgame --cov-report=xml
+          --ignore=tests/test_oblique_forest.py
+          --ignore=tests/test_grande.py
+          --ignore=tests/test_tabdpt.py
+          --ignore=tests/test_nlp.py
+          --ignore=tests/test_nlp_translation.py
+          --ignore=tests/test_vision.py
+          -k "not TestExtendedIsolationForest and not (TestTabularPredictor and (test_predict_classification or test_predict_proba_classification or test_leaderboard)) and not (TestAutoMLPredictor and (test_predict_automl or test_chained_fit_predict)) and not (TestCORELS and test_fit_predict_without_corels) and not (TestGAM and test_partial_dependence) and not TestUMAPReducer and not TestTriMAPReducer and not TestPHATEReducer and not TestPaCMAPReducer and not TestTextPredictor"
+
+      - name: Upload coverage
+        if: matrix.python-version == '3.11'
+        uses: codecov/codecov-action@v4
+        with:
+          files: coverage.xml
+          fail_ci_if_error: false
+        env:
+          CODECOV_TOKEN: ${{ secrets.CODECOV_TOKEN }}
+
+  typecheck:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v4
+
+      - uses: actions/setup-python@v5
+        with:
+          python-version: "3.11"
+          cache: pip
+
+      - name: Install dependencies
+        run: pip install -e ".[dev,tabular,calibration,explain,fairness]"
+
+      - name: Type check (informational)
+        run: |
+          echo "Running mypy (informational — errors do not block CI)"
+          mypy endgame/ --ignore-missing-imports --no-error-summary 2>&1 | tail -20 || true
+          echo ""
+          mypy endgame/ --ignore-missing-imports 2>&1 | grep -c "error:" || echo "0 errors"
+          echo "See full output above. Typecheck is informational until error count reaches zero."

endgame_ml-1.0.0/.github/workflows/publish.yml

@@ -0,0 +1,28 @@
+name: Publish to PyPI
+
+on:
+  release:
+    types: [published]
+
+permissions:
+  id-token: write
+
+jobs:
+  build-and-publish:
+    runs-on: ubuntu-latest
+    environment: pypi
+    steps:
+      - uses: actions/checkout@v4
+
+      - uses: actions/setup-python@v5
+        with:
+          python-version: "3.12"
+
+      - name: Install build tools
+        run: pip install hatch
+
+      - name: Build package
+        run: hatch build
+
+      - name: Publish to PyPI
+        uses: pypa/gh-action-pypi-publish@release/v1

endgame_ml-1.0.0/.gitignore

@@ -0,0 +1,77 @@
+# Python
+__pycache__/
+*.py[cod]
+*$py.class
+*.so
+*.egg-info/
+dist/
+build/
+*.egg
+
+# Virtual environments
+.venv/
+venv/
+env/
+
+# IDE
+.idea/
+.vscode/
+*.swp
+*.swo
+*~
+
+# OS files
+.DS_Store
+Thumbs.db
+._*
+
+# Claude Code
+.claude/
+
+# Build caches (e.g. compiled CORELS binary)
+.cache/
+
+# Rust build artifacts (C5.0, Cubist)
+**/target/
+
+# Git backup
+.git-backup/
+
+# Logs and artifacts
+*.log
+lightning_logs/
+catboost_info/
+
+# Data and results (large files)
+data/
+results/
+saved_models/
+submissions/
+*.parquet
+
+# Benchmark outputs
+benchmark_results_report.html
+*.csv
+
+# Sphinx build output
+docs/_build/
+
+# Coverage
+.coverage
+.coverage.*
+htmlcov/
+
+# PyTorch temp
+.pt_tmp/
+
+# Paper (unpublished research — not public yet)
+paper/
+
+# AutoML output directories
+automl_best_quality_output/
+automl_interpretable_output/
+automl_evolutionary_output/
+
+# Requirements files (use pyproject.toml)
+requirements*.txt
+!docs/requirements.txt

endgame_ml-1.0.0/.mcp.json

@@ -0,0 +1,8 @@
+{
+  "mcpServers": {
+    "endgame": {
+      "command": "python",
+      "args": ["-m", "endgame.mcp"]
+    }
+  }
+}

endgame_ml-1.0.0/.readthedocs.yml

@@ -0,0 +1,15 @@
+version: 2
+
+build:
+  os: ubuntu-22.04
+  tools:
+    python: "3.12"
+
+sphinx:
+  configuration: docs/conf.py
+
+python:
+  install:
+    - requirements: docs/requirements.txt
+    - method: pip
+      path: .

endgame_ml-1.0.0/CHANGELOG.md

@@ -0,0 +1,44 @@
+# Changelog
+
+All notable changes to this project will be documented in this file.
+
+The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.1.0/),
+and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
+
+## [1.0.0] - 2026-02-22
+
+First stable release of Endgame.
+
+### Changed
+- Version bump from 0.7.0-alpha to 1.0.0
+- Updated development status to Production/Stable
+
+### Fixed
+- `eg.timeseries`, `eg.signal`, `eg.automl`, `eg.dimensionality_reduction`, `eg.feature_selection` now accessible via top-level lazy loading
+- Updated `__all__` to include all public modules
+
+## [0.7.0-alpha] - 2026-02-19
+
+Initial public release of Endgame.
+
+### Added
+- **100+ models** with unified sklearn-compatible API: GBDTs (LightGBM, XGBoost, CatBoost), deep tabular (FT-Transformer, SAINT, NODE, TabPFN, NAM, GANDALF), custom trees (Rotation Forest, C5.0/Cubist, Oblique, Evolutionary), rules (RuleFit, FURIA), Bayesian classifiers (TAN, KDB, ESKDB), kernel methods, probabilistic models (NGBoost, BART), and interpretable models (EBM, MARS)
+- **Polars-powered preprocessing**: SafeTargetEncoder, AutoAggregator, 18+ resampling methods (SMOTE family, ADASYN, geometric, generative), MICE/MissForest imputation, ConfidentLearning noise detection
+- **Ensemble methods**: SuperLearner (NNLS), HillClimbingEnsemble, StackingEnsemble, BlendingEnsemble, ThresholdOptimizer, knowledge distillation
+- **Calibration**: Conformal prediction (classification + regression), CQR, Venn-ABERS, Temperature/Platt/Beta/Isotonic scaling
+- **Validation**: AdversarialValidator, PurgedTimeSeriesSplit, StratifiedGroupKFold, CPCV
+- **42 interactive visualizations**: ROC, PR, calibration, PDP/ICE, waterfall, confusion matrix, Sankey, network diagrams, tree visualizer, and more — all self-contained HTML with no CDN dependencies
+- **Signal processing**: filtering (Butterworth, FIR, notch), spectral analysis (FFT, Welch, multitaper), wavelets (CWT, DWT), entropy measures, complexity measures, spatial filtering (CSP)
+- **Time series**: statistical forecasters (AutoARIMA, AutoETS, MSTL), neural forecasters (N-BEATS, TFT, PatchTST), ROCKET/Hydra classification
+- **Explainability**: SHAP, LIME, PDP, feature interactions, counterfactual explanations
+- **Fairness**: demographic parity, equalized odds, bias mitigation (Reweighing, Exponentiated Gradient, Calibrated EqOdds)
+- **AutoML**: TabularPredictor with preset system (best_quality, high_quality, good_quality, medium_quality, fast, interpretable) with evolutionary/genetic search strategy
+- **Anomaly detection**: Isolation Forest, Extended IF, LOF, GritBot, PyOD wrapper (39+ algorithms)
+- **Persistence**: model save/load, ONNX export, ModelServer for inference
+- **MCP server**: 20 tools + 6 resources for LLM-powered ML pipelines with consistent categorical encoding, timeout protection, and structured error handling
+- **Benchmark suite**: OpenML integration, meta-learning, synthetic data generation
+- **Example notebooks**: 6 notebooks covering quickstart, interpretable models, AutoML, ensembles, MCP, and signal/timeseries
+
+### Dependencies
+- Core: numpy>=1.24, polars>=0.20, scikit-learn>=1.3, optuna>=3.4, scipy>=1.10, networkx>=3.0
+- Optional groups: `tabular`, `vision`, `nlp`, `audio`, `benchmark`, `calibration`, `explain`, `fairness`, `deployment`, `mcp`

endgame_ml-1.0.0/CITATION.cff

@@ -0,0 +1,34 @@
+cff-version: 1.2.0
+title: "Endgame: A Living Library of Machine Learning"
+message: "If you use this software, please cite it as below."
+type: software
+authors:
+  - given-names: Cameron Hamilton
+    family-names: Hamilton
+    affiliation: Alliance AI
+repository-code: "https://github.com/allianceai/endgame"
+url: "https://github.com/allianceai/endgame"
+license: Apache-2.0
+version: 1.0.0
+date-released: "2026-02-22"
+keywords:
+  - machine-learning
+  - deep-learning
+  - tabular-data
+  - ensemble-methods
+  - automl
+  - time-series
+  - signal-processing
+  - interpretability
+  - calibration
+  - scikit-learn
+abstract: >-
+  Endgame brings together classic and state-of-the-art machine learning
+  methods from every domain under a single scikit-learn-compatible API.
+  It provides 300+ estimators, transformers, and visualizers across
+  tabular, time series, signal processing, vision, NLP, and audio —
+  including deep tabular models (FT-Transformer, SAINT, TabPFN),
+  time series classification (ROCKET, HYDRA), conformal prediction
+  with coverage guarantees, 22 ensemble strategies, and 42 self-contained
+  interactive HTML visualizations. Designed for research, production,
+  and competition.

endgame_ml-1.0.0/CODE_OF_CONDUCT.md

@@ -0,0 +1,54 @@
+# Contributor Covenant Code of Conduct
+
+## Our Pledge
+
+We as members, contributors, and leaders pledge to make participation in our
+community a welcoming experience for everyone, regardless of background or
+identity.
+
+We pledge to act and interact in ways that contribute to an open, welcoming,
+diverse, inclusive, and healthy community.
+
+## Our Standards
+
+Examples of behavior that contributes to a positive environment:
+
+- Using welcoming and inclusive language
+- Being respectful of differing viewpoints and experiences
+- Gracefully accepting constructive criticism
+- Focusing on what is best for the community
+- Showing empathy towards other community members
+
+Examples of unacceptable behavior:
+
+- Trolling, insulting or derogatory comments, and personal attacks
+- Public or private communication that others would reasonably find unwelcome
+- Publishing others' private information without explicit permission
+- Other conduct which could reasonably be considered inappropriate in a
+  professional setting
+
+## Enforcement Responsibilities
+
+Community leaders are responsible for clarifying and enforcing our standards of
+acceptable behavior and will take appropriate and fair corrective action in
+response to any behavior that they deem inappropriate or harmful.
+
+## Scope
+
+This Code of Conduct applies within all community spaces, and also applies when
+an individual is officially representing the community in public spaces.
+
+## Enforcement
+
+Instances of unacceptable behavior may be reported to the community leaders
+responsible for enforcement via the contact methods listed in the repository.
+
+All complaints will be reviewed and investigated promptly and fairly. Community
+leaders are obligated to respect the privacy and security of the reporter of
+any incident.
+
+## Attribution
+
+This Code of Conduct is adapted from the [Contributor Covenant](https://www.contributor-covenant.org),
+version 2.1, available at
+[https://www.contributor-covenant.org/version/2/1/code_of_conduct.html](https://www.contributor-covenant.org/version/2/1/code_of_conduct.html).