endgame-ml 1.0.0__py3-none-any.whl

endgame/__init__.py

@@ -0,0 +1,95 @@
+"""
+Endgame: The Competitive Machine Learning Framework
+
+Sklearn-Native, Polars-Powered, Victory-Driven.
+
+Usage:
+    import endgame as eg
+
+    # Check for train/test drift
+    av = eg.validation.AdversarialValidator()
+    result = av.check_drift(X_train, X_test)
+
+    # Feature engineering
+    encoder = eg.preprocessing.SafeTargetEncoder()
+    X_encoded = encoder.fit_transform(X, y)
+
+    # Train models
+    model = eg.models.LGBMWrapper(preset='endgame')
+    model.fit(X_train, y_train)
+
+    # Ensemble
+    ensemble = eg.ensemble.HillClimbingEnsemble(metric='roc_auc')
+    ensemble.fit(oof_predictions, y_train)
+"""
+
+__version__ = "1.0.0"
+
+# Note: models imported lazily for heavyweight optional dependencies
+# from endgame import models
+from endgame import (
+    anomaly,
+    calibration,
+    clustering,
+    ensemble,
+    fairness,
+    preprocessing,
+    semi_supervised,
+    tune,
+    utils,
+    validation,
+)
+
+
+# Lazy imports for heavy modules
+def __getattr__(name: str):
+    import importlib
+
+    # Models and other heavy modules imported lazily for optional dependencies
+    if name in ("models", "vision", "nlp", "audio", "benchmark", "kaggle", "quick", "visualization", "persistence", "explain", "tracking", "timeseries", "signal", "automl", "dimensionality_reduction", "feature_selection"):
+        module = importlib.import_module(f"endgame.{name}")
+        globals()[name] = module
+        return module
+    if name == "save":
+        from endgame.persistence import save
+        return save
+    if name == "load":
+        from endgame.persistence import load
+        return load
+    if name == "export_onnx":
+        from endgame.persistence import export_onnx
+        return export_onnx
+    raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
+
+__all__ = [
+    "__version__",
+    "validation",
+    "preprocessing",
+    "models",
+    "ensemble",
+    "tune",
+    "calibration",
+    "semi_supervised",
+    "anomaly",
+    "fairness",
+    "clustering",
+    "benchmark",
+    "vision",
+    "nlp",
+    "audio",
+    "kaggle",
+    "utils",
+    "quick",
+    "visualization",
+    "persistence",
+    "tracking",
+    "timeseries",
+    "signal",
+    "automl",
+    "dimensionality_reduction",
+    "feature_selection",
+    "save",
+    "load",
+    "export_onnx",
+    "explain",
+]

endgame/anomaly/__init__.py

@@ -0,0 +1,58 @@
+"""Anomaly detection module: Isolation Forest, LOF, GritBot, PyOD integration, and more.
+
+This module provides sklearn-compatible anomaly detectors with competition-tuned
+defaults, including wrappers for popular algorithms and PyOD integration.
+
+Example
+-------
+>>> from endgame.anomaly import IsolationForestDetector, PyODDetector, GritBotDetector
+>>>
+>>> # Simple Isolation Forest
+>>> detector = IsolationForestDetector(contamination=0.1)
+>>> detector.fit(X_train)
+>>> scores = detector.decision_function(X_test)  # Higher = more anomalous
+>>> labels = detector.predict(X_test)  # 1 = anomaly, 0 = normal
+>>>
+>>> # PyOD wrapper with any algorithm
+>>> detector = PyODDetector(algorithm='ECOD', contamination=0.05)
+>>> detector.fit(X_train)
+>>> scores = detector.decision_function(X_test)
+>>>
+>>> # GritBot: Rule-based with interpretable context
+>>> detector = GritBotDetector(filtering_level=50)
+>>> detector.fit(X_train)
+>>> print(detector.get_anomaly_report())  # Human-readable explanations
+"""
+
+from endgame.anomaly.gritbot import (
+    Anomaly,
+    AnomalyContext,
+    GritBotDetector,
+)
+from endgame.anomaly.isolation_forest import IsolationForestDetector
+
+try:
+    from treeple import ExtendedIsolationForest
+except ImportError:
+    from endgame.anomaly.isolation_forest import ExtendedIsolationForest
+from endgame.anomaly.lof import LocalOutlierFactorDetector
+from endgame.anomaly.pyod_wrapper import (
+    PYOD_ALGORITHMS,
+    PyODDetector,
+    create_detector_ensemble,
+)
+
+__all__ = [
+    # Core detectors
+    "IsolationForestDetector",
+    "ExtendedIsolationForest",
+    "LocalOutlierFactorDetector",
+    # GritBot (rule-based, interpretable)
+    "GritBotDetector",
+    "Anomaly",
+    "AnomalyContext",
+    # PyOD integration
+    "PyODDetector",
+    "PYOD_ALGORITHMS",
+    "create_detector_ensemble",
+]