deep-inference 0.1.0__tar.gz

deep_inference-0.1.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2024 Pranjal Rawat
+
+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.

deep_inference-0.1.0/MANIFEST.in

@@ -0,0 +1,25 @@
+# Include
+include LICENSE
+include README.md
+include pyproject.toml
+
+# Include source
+recursive-include src *.py *.typed
+
+# Exclude everything else
+prune archive
+prune data
+prune docs
+prune logs
+prune paper
+prune prototypes
+prune references
+prune tests
+prune src2
+global-exclude *.csv
+global-exclude *.pkl
+global-exclude *.json
+global-exclude *.log
+global-exclude __pycache__
+global-exclude *.pyc
+global-exclude .DS_Store

deep_inference-0.1.0/PKG-INFO

@@ -0,0 +1,209 @@
+Metadata-Version: 2.4
+Name: deep-inference
+Version: 0.1.0
+Summary: Production-grade econometrics with neural networks
+Author: Pranjal Rawat
+License-Expression: MIT
+Project-URL: Homepage, https://github.com/rawatpranjal/deep-inference
+Project-URL: Repository, https://github.com/rawatpranjal/deep-inference
+Project-URL: Documentation, https://rawatpranjal.github.io/deep-inference
+Keywords: econometrics,causal-inference,deep-learning,neural-networks,statistics,machine-learning
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Science/Research
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Classifier: Topic :: Scientific/Engineering :: Mathematics
+Classifier: Typing :: Typed
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: torch>=2.0
+Requires-Dist: numpy>=1.24
+Requires-Dist: pandas>=2.0
+Requires-Dist: scipy>=1.10
+Requires-Dist: scikit-learn>=1.3
+Requires-Dist: formulaic>=1.0
+Requires-Dist: tabulate>=0.9
+Requires-Dist: tqdm>=4.65
+Provides-Extra: dev
+Requires-Dist: pytest>=8.0; extra == "dev"
+Requires-Dist: pytest-cov>=4.0; extra == "dev"
+Requires-Dist: ruff>=0.4; extra == "dev"
+Requires-Dist: mypy>=1.10; extra == "dev"
+Requires-Dist: pre-commit>=3.7; extra == "dev"
+Provides-Extra: parity
+Requires-Dist: rpy2>=3.5; extra == "parity"
+Provides-Extra: docs
+Requires-Dist: sphinx>=7.0; extra == "docs"
+Requires-Dist: pydata-sphinx-theme>=0.16; extra == "docs"
+Requires-Dist: myst-parser>=2.0; extra == "docs"
+Requires-Dist: sphinx-autodoc-typehints>=1.25; extra == "docs"
+Requires-Dist: sphinx-copybutton>=0.5; extra == "docs"
+Provides-Extra: plotting
+Requires-Dist: matplotlib>=3.8; extra == "plotting"
+Requires-Dist: seaborn>=0.13; extra == "plotting"
+Provides-Extra: all
+Requires-Dist: deep-inference[dev,docs,parity,plotting]; extra == "all"
+Dynamic: license-file
+
+# `deep-inference`
+
+**Deep Learning for Individual Heterogeneity**
+
+`deep-inference` is a Python package for **enriching structural economic models** with deep learning. It implements the framework developed by Farrell, Liang, and Misra (2021, 2025) to recover rich, non-linear parameter heterogeneity ($\theta(X)$) while maintaining the interpretability and validity of structural economics.
+
+Standard deep learning minimizes prediction error, which leads to biased parameter estimates ("The Inference Trap"). This package implements **Influence Function-based Debiasing** (a form of Double Machine Learning) to provide valid confidence intervals and p-values for economic targets.
+1
+[![Python 3.10+](https://img.shields.io/badge/python-3.10+-blue.svg)](https://www.python.org/downloads/)
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
+
+## Why `deep-inference`?
+
+Economic structure and Machine Learning are **complements**, not substitutes.
+*   **Deep Learning** provides the capacity to learn complex, high-dimensional heterogeneity.
+*   **Structural Models** provide the constraints necessary for causal interpretation and counterfactuals.
+
+`deep-inference` enforces the structural loss (e.g., Tobit likelihood) on the output of a neural network, ensuring that the estimated parameters $\alpha(X), \beta(X)$ respect the economic theory.
+
+## Installation
+
+```bash
+pip install deep-inference
+```
+
+## Quickstart
+
+Estimate a linear demand model where Price Elasticity $\beta$ varies non-linearly with customer characteristics $X$.
+
+```python
+import torch
+from deep_inference import get_dgp, get_family, influence
+
+# 1. Generate synthetic data (Linear Demand with Heterogeneity)
+dgp = get_dgp("linear", seed=42)
+data = dgp.generate(n=2000)
+
+# 2. Define the Structural Family
+# (Defines the Loss, Score, and Hessian automatically)
+family = get_family("linear")
+
+# 3. Run Inference
+# The package trains the structural network, computes the influence function,
+# and aggregates results via 10-fold cross-fitting.
+result = influence(
+    X=data.X, T=data.T, Y=data.Y,
+    family=family,
+    config={
+        "hidden_dims": [128, 64, 32],
+        "epochs": 50,
+        "lr": 0.01
+    }
+)
+
+print(f"Avg Elasticity (Truth): {data.mu_true:.4f}")
+print(f"Avg Elasticity (Est):   {result.mu_hat:.4f}")
+print(f"Standard Error:         {result.se:.4f}")
+print(f"95% CI:                 [{result.ci_lower:.4f}, {result.ci_upper:.4f}]")
+```
+
+## Supported Structural Families
+
+The package abstracts the math of influence functions. You simply select the family that matches your outcome variable.
+
+| Family | Model Structure | Use Case |
+|--------|-----------------|----------|
+| **Linear** | $Y = \alpha(X) + \beta(X)T + \varepsilon$ | Wages, Test Scores, consumption |
+| **Logit** | $P(Y=1) = \sigma(\alpha(X) + \beta(X)T)$ | Binary Choice, Market Entry |
+| **Tobit** | $Y = \max(0, \alpha(X) + \beta(X)T + \varepsilon)$ | Labor Supply, Censored Demand |
+| **Poisson** | $Y \sim \text{Pois}(\exp(\alpha(X) + \beta(X)T))$ | Patent Counts, Doctor Visits |
+| **Gamma** | $Y \sim \text{Gamma}(k(X), \theta(X))$ | Healthcare Costs, Insurance Claims |
+| **Weibull** | $Y \sim \text{Weibull}(\dots)$ | Duration Analysis, Unemployment Spells |
+
+*Note: For complex models like Tobit, `deep-inference` automatically handles the joint estimation of structural variance $\sigma(X)$ required for consistent inference.*
+
+## Methodological Details
+
+### 1. The Enriched Model
+We replace fixed parameters $\theta$ with neural networks $\theta(X)$:
+
+```math
+\hat{\theta}(\cdot) = \arg \min_{\theta \in \mathcal{F}_{DNN}} \sum \ell(y_i, t_i, \theta(x_i))
+```
+
+### 2. The Influence Function Correction
+Naive averaging of $\hat{\theta}(X)$ yields biased inference. We construct a Neyman-Orthogonal score $\psi$ using the **Influence Function**:
+
+```math
+\psi(z) = H(\hat{\theta}) + \nabla_\theta H \cdot \Lambda(x)^{-1} \cdot \nabla_\theta \ell(z, \hat{\theta})
+```
+
+Where $\Lambda(x) = \mathbb{E}[\nabla^2 \ell \mid X=x]$ is the conditional Hessian.
+*   **Automatic Differentiation:** `deep-inference` uses PyTorch Autograd to compute exact Jacobians and Hessians for any model family.
+*   **Stability:** Includes Tikhonov regularization for inverting Hessians in non-linear models (e.g., Logit/Tobit).
+
+## Validation (Monte Carlo Results)
+
+M=30 simulations, N=10,000 observations. Target: 95% coverage, SE ratio ≈ 1.0.
+
+### Linear Model
+
+![Linear Results](logs/kde_money_slide.png)
+
+| Config | K | Network | Coverage | SE Ratio | RMSE | Bias²/MSE |
+|--------|---|---------|----------|----------|------|-----------|
+| **Best** | 50 | [64,32] | **93.3%** | 1.03 | **0.032** | 22% |
+| Deep | 20 | [128,64,32] | 93.3% | 1.02 | 0.033 | 21% |
+| E=100 | 20 | [64,32] | 90.0% | 0.90 | 0.036 | 18% |
+| Separate | 20 | [128,64,32]×2 | 80.0% | 0.82 | 0.036 | 14% |
+| Naive | — | — | 10% | 0.09 | 0.083 | 1% |
+
+**Finding:** K=50 folds achieves best coverage and lowest RMSE. Separate networks don't help.
+
+### Logit Model
+
+![Logit Results](logs/logit_stress_test/logit_results.png)
+
+Target: E[β(X)] = average log-odds ratio
+
+| Method | Coverage | SE Ratio | RMSE | Hessian min λ |
+|--------|----------|----------|------|---------------|
+| **Influence** | **90.0%** | 0.93 | **0.054** | 0.051 |
+| Naive | 3.3% | 0.03 | 0.108 | — |
+
+### Summary
+
+| Model | Target | Naive Cov | IF Cov | RMSE Improvement |
+|-------|--------|-----------|--------|------------------|
+| Linear | E[β(X)] | 10% | **93%** | 2.6× |
+| Logit | E[β(X)] | 3% | **90%** | 2.0× |
+| Poisson | E[β(X)] | TBD | TBD | — |
+| Gamma | E[β(X)] | TBD | TBD | — |
+
+## Citation
+
+```bibtex
+@article{farrell2021deep,
+  title={Deep Neural Networks for Estimation and Inference},
+  author={Farrell, Max H. and Liang, Tengyuan and Misra, Sanjog},
+  journal={Econometrica},
+  volume={89},
+  number={1},
+  pages={181--213},
+  year={2021}
+}
+
+@article{farrell2025heterogeneity,
+  title={Deep Learning for Individual Heterogeneity},
+  author={Farrell, Max H. and Liang, Tengyuan and Misra, Sanjog},
+  journal={Working Paper},
+  year={2025}
+}
+```
+
+## License
+
+MIT

deep_inference-0.1.0/README.md

@@ -0,0 +1,157 @@
+# `deep-inference`
+
+**Deep Learning for Individual Heterogeneity**
+
+`deep-inference` is a Python package for **enriching structural economic models** with deep learning. It implements the framework developed by Farrell, Liang, and Misra (2021, 2025) to recover rich, non-linear parameter heterogeneity ($\theta(X)$) while maintaining the interpretability and validity of structural economics.
+
+Standard deep learning minimizes prediction error, which leads to biased parameter estimates ("The Inference Trap"). This package implements **Influence Function-based Debiasing** (a form of Double Machine Learning) to provide valid confidence intervals and p-values for economic targets.
+1
+[![Python 3.10+](https://img.shields.io/badge/python-3.10+-blue.svg)](https://www.python.org/downloads/)
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
+
+## Why `deep-inference`?
+
+Economic structure and Machine Learning are **complements**, not substitutes.
+*   **Deep Learning** provides the capacity to learn complex, high-dimensional heterogeneity.
+*   **Structural Models** provide the constraints necessary for causal interpretation and counterfactuals.
+
+`deep-inference` enforces the structural loss (e.g., Tobit likelihood) on the output of a neural network, ensuring that the estimated parameters $\alpha(X), \beta(X)$ respect the economic theory.
+
+## Installation
+
+```bash
+pip install deep-inference
+```
+
+## Quickstart
+
+Estimate a linear demand model where Price Elasticity $\beta$ varies non-linearly with customer characteristics $X$.
+
+```python
+import torch
+from deep_inference import get_dgp, get_family, influence
+
+# 1. Generate synthetic data (Linear Demand with Heterogeneity)
+dgp = get_dgp("linear", seed=42)
+data = dgp.generate(n=2000)
+
+# 2. Define the Structural Family
+# (Defines the Loss, Score, and Hessian automatically)
+family = get_family("linear")
+
+# 3. Run Inference
+# The package trains the structural network, computes the influence function,
+# and aggregates results via 10-fold cross-fitting.
+result = influence(
+    X=data.X, T=data.T, Y=data.Y,
+    family=family,
+    config={
+        "hidden_dims": [128, 64, 32],
+        "epochs": 50,
+        "lr": 0.01
+    }
+)
+
+print(f"Avg Elasticity (Truth): {data.mu_true:.4f}")
+print(f"Avg Elasticity (Est):   {result.mu_hat:.4f}")
+print(f"Standard Error:         {result.se:.4f}")
+print(f"95% CI:                 [{result.ci_lower:.4f}, {result.ci_upper:.4f}]")
+```
+
+## Supported Structural Families
+
+The package abstracts the math of influence functions. You simply select the family that matches your outcome variable.
+
+| Family | Model Structure | Use Case |
+|--------|-----------------|----------|
+| **Linear** | $Y = \alpha(X) + \beta(X)T + \varepsilon$ | Wages, Test Scores, consumption |
+| **Logit** | $P(Y=1) = \sigma(\alpha(X) + \beta(X)T)$ | Binary Choice, Market Entry |
+| **Tobit** | $Y = \max(0, \alpha(X) + \beta(X)T + \varepsilon)$ | Labor Supply, Censored Demand |
+| **Poisson** | $Y \sim \text{Pois}(\exp(\alpha(X) + \beta(X)T))$ | Patent Counts, Doctor Visits |
+| **Gamma** | $Y \sim \text{Gamma}(k(X), \theta(X))$ | Healthcare Costs, Insurance Claims |
+| **Weibull** | $Y \sim \text{Weibull}(\dots)$ | Duration Analysis, Unemployment Spells |
+
+*Note: For complex models like Tobit, `deep-inference` automatically handles the joint estimation of structural variance $\sigma(X)$ required for consistent inference.*
+
+## Methodological Details
+
+### 1. The Enriched Model
+We replace fixed parameters $\theta$ with neural networks $\theta(X)$:
+
+```math
+\hat{\theta}(\cdot) = \arg \min_{\theta \in \mathcal{F}_{DNN}} \sum \ell(y_i, t_i, \theta(x_i))
+```
+
+### 2. The Influence Function Correction
+Naive averaging of $\hat{\theta}(X)$ yields biased inference. We construct a Neyman-Orthogonal score $\psi$ using the **Influence Function**:
+
+```math
+\psi(z) = H(\hat{\theta}) + \nabla_\theta H \cdot \Lambda(x)^{-1} \cdot \nabla_\theta \ell(z, \hat{\theta})
+```
+
+Where $\Lambda(x) = \mathbb{E}[\nabla^2 \ell \mid X=x]$ is the conditional Hessian.
+*   **Automatic Differentiation:** `deep-inference` uses PyTorch Autograd to compute exact Jacobians and Hessians for any model family.
+*   **Stability:** Includes Tikhonov regularization for inverting Hessians in non-linear models (e.g., Logit/Tobit).
+
+## Validation (Monte Carlo Results)
+
+M=30 simulations, N=10,000 observations. Target: 95% coverage, SE ratio ≈ 1.0.
+
+### Linear Model
+
+![Linear Results](logs/kde_money_slide.png)
+
+| Config | K | Network | Coverage | SE Ratio | RMSE | Bias²/MSE |
+|--------|---|---------|----------|----------|------|-----------|
+| **Best** | 50 | [64,32] | **93.3%** | 1.03 | **0.032** | 22% |
+| Deep | 20 | [128,64,32] | 93.3% | 1.02 | 0.033 | 21% |
+| E=100 | 20 | [64,32] | 90.0% | 0.90 | 0.036 | 18% |
+| Separate | 20 | [128,64,32]×2 | 80.0% | 0.82 | 0.036 | 14% |
+| Naive | — | — | 10% | 0.09 | 0.083 | 1% |
+
+**Finding:** K=50 folds achieves best coverage and lowest RMSE. Separate networks don't help.
+
+### Logit Model
+
+![Logit Results](logs/logit_stress_test/logit_results.png)
+
+Target: E[β(X)] = average log-odds ratio
+
+| Method | Coverage | SE Ratio | RMSE | Hessian min λ |
+|--------|----------|----------|------|---------------|
+| **Influence** | **90.0%** | 0.93 | **0.054** | 0.051 |
+| Naive | 3.3% | 0.03 | 0.108 | — |
+
+### Summary
+
+| Model | Target | Naive Cov | IF Cov | RMSE Improvement |
+|-------|--------|-----------|--------|------------------|
+| Linear | E[β(X)] | 10% | **93%** | 2.6× |
+| Logit | E[β(X)] | 3% | **90%** | 2.0× |
+| Poisson | E[β(X)] | TBD | TBD | — |
+| Gamma | E[β(X)] | TBD | TBD | — |
+
+## Citation
+
+```bibtex
+@article{farrell2021deep,
+  title={Deep Neural Networks for Estimation and Inference},
+  author={Farrell, Max H. and Liang, Tengyuan and Misra, Sanjog},
+  journal={Econometrica},
+  volume={89},
+  number={1},
+  pages={181--213},
+  year={2021}
+}
+
+@article{farrell2025heterogeneity,
+  title={Deep Learning for Individual Heterogeneity},
+  author={Farrell, Max H. and Liang, Tengyuan and Misra, Sanjog},
+  journal={Working Paper},
+  year={2025}
+}
+```
+
+## License
+
+MIT

deep_inference-0.1.0/pyproject.toml

@@ -0,0 +1,139 @@
+[build-system]
+requires = ["setuptools>=61.0", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "deep-inference"
+version = "0.1.0"
+description = "Production-grade econometrics with neural networks"
+readme = "README.md"
+license = "MIT"
+authors = [
+    {name = "Pranjal Rawat"}
+]
+requires-python = ">=3.10"
+classifiers = [
+    "Development Status :: 4 - Beta",
+    "Intended Audience :: Science/Research",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+    "Programming Language :: Python :: 3.13",
+    "Topic :: Scientific/Engineering :: Artificial Intelligence",
+    "Topic :: Scientific/Engineering :: Mathematics",
+    "Typing :: Typed",
+]
+keywords = [
+    "econometrics",
+    "causal-inference",
+    "deep-learning",
+    "neural-networks",
+    "statistics",
+    "machine-learning",
+]
+
+dependencies = [
+    "torch>=2.0",
+    "numpy>=1.24",
+    "pandas>=2.0",
+    "scipy>=1.10",
+    "scikit-learn>=1.3",
+    "formulaic>=1.0",
+    "tabulate>=0.9",
+    "tqdm>=4.65",
+]
+
+[project.optional-dependencies]
+dev = [
+    "pytest>=8.0",
+    "pytest-cov>=4.0",
+    "ruff>=0.4",
+    "mypy>=1.10",
+    "pre-commit>=3.7",
+]
+parity = [
+    "rpy2>=3.5",
+]
+docs = [
+    "sphinx>=7.0",
+    "pydata-sphinx-theme>=0.16",
+    "myst-parser>=2.0",
+    "sphinx-autodoc-typehints>=1.25",
+    "sphinx-copybutton>=0.5",
+]
+plotting = [
+    "matplotlib>=3.8",
+    "seaborn>=0.13",
+]
+all = [
+    "deep-inference[dev,parity,docs,plotting]",
+]
+
+[project.urls]
+Homepage = "https://github.com/rawatpranjal/deep-inference"
+Repository = "https://github.com/rawatpranjal/deep-inference"
+Documentation = "https://rawatpranjal.github.io/deep-inference"
+
+[tool.setuptools.packages.find]
+where = ["src"]
+
+[tool.setuptools.package-data]
+deep_inference = ["py.typed"]
+
+[tool.ruff]
+line-length = 88
+target-version = "py310"
+src = ["src"]
+
+[tool.ruff.lint]
+select = [
+    "E",      # pycodestyle errors
+    "W",      # pycodestyle warnings
+    "F",      # Pyflakes
+    "I",      # isort
+    "B",      # flake8-bugbear
+    "C4",     # flake8-comprehensions
+    "UP",     # pyupgrade
+    "NPY",    # NumPy-specific rules
+]
+ignore = [
+    "E501",   # line too long (handled by formatter)
+]
+
+[tool.ruff.lint.isort]
+known-first-party = ["deep_inference"]
+
+[tool.mypy]
+python_version = "3.10"
+warn_return_any = true
+warn_unused_ignores = true
+strict = true
+plugins = ["numpy.typing.mypy_plugin"]
+
+[[tool.mypy.overrides]]
+module = [
+    "torch.*",
+    "scipy.*",
+    "sklearn.*",
+    "formulaic.*",
+    "tabulate.*",
+    "tqdm.*",
+]
+ignore_missing_imports = true
+
+[tool.pytest.ini_options]
+testpaths = ["tests"]
+python_files = ["test_*.py"]
+addopts = "-v --tb=short"
+
+[tool.coverage.run]
+source = ["src/deep_inference"]
+branch = true
+
+[tool.coverage.report]
+exclude_lines = [
+    "pragma: no cover",
+    "if TYPE_CHECKING:",
+    "raise NotImplementedError",
+]

deep_inference-0.1.0/setup.cfg

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

deep_inference-0.1.0/src/deep_inference/__init__.py

@@ -0,0 +1,46 @@
+"""deep_inference: Influence function validation for neural network inference.
+
+Implements the Farrell, Liang, Misra (2021, 2025) approach for valid inference
+with neural network estimators. NOT DML - neural nets output structural
+parameters directly and influence functions correct for regularization bias.
+
+Usage:
+    from deep_inference import get_dgp, get_family, naive, influence, bootstrap
+
+    # Generate data
+    dgp = get_dgp("linear")
+    data = dgp.generate(1000)
+
+    # Get family for loss/residual/weights
+    family = get_family("linear")
+
+    # Run inference
+    mu_hat, se = influence(data.X, data.T, data.Y, family, config)
+"""
+
+from .dgp import get_dgp, DGPS, verify_ground_truth
+from .families import get_family, FAMILIES
+from .models import StructuralNet, NuisanceNet, train_structural, train_nuisance
+from .inference import naive, influence, bootstrap, METHODS
+from .metrics import compute_metrics, print_table
+from .real_data import load_dataset, DATASETS, RealDataResult
+from .classical import classical_logit, classical_poisson, classical_ols, ClassicalResult
+
+__all__ = [
+    # DGP
+    "get_dgp", "DGPS", "verify_ground_truth",
+    # Families
+    "get_family", "FAMILIES",
+    # Models
+    "StructuralNet", "NuisanceNet", "train_structural", "train_nuisance",
+    # Inference
+    "naive", "influence", "bootstrap", "METHODS",
+    # Metrics
+    "compute_metrics", "print_table",
+    # Real data
+    "load_dataset", "DATASETS", "RealDataResult",
+    # Classical estimators
+    "classical_logit", "classical_poisson", "classical_ols", "ClassicalResult",
+]
+
+__version__ = "2.0.0"