python source code of mediator

whynot-master
- roadmap.txt
- whynot
  - gym
    - utils
      - __init__.py
    - __init__.py
    - spaces
      - __init__.py
    - envs
      - registration.py
      - ode_env.py
      - __init__.py
      - README.md
  - framework.py
  - dynamics.py
  - algorithms
    - causal_suite.py
    - ols.py
    - propensity_score_matching.py
    - __init__.py
    - propensity_weighted_ols.py
  - causal_graphs.py
  - __init__.py
  - utils.py
  - simulators
    - opioid
      - environments.py
      - .experiments.py.swo
      - experiments.py
      - simulator.py
      - __init__.py
      - README.md
    - lalonde
      - lalonde.csv
      - experiments.py
      - __init__.py
      - README.md
    - civil_violence
      - experiments.py
      - simulator.py
      - model.py
      - __init__.py
      - README.md
    - world3
      - environments.py
      - experiments.py
      - simulator.py
      - world3_app.js
      - __init__.py
      - README.md
    - dice
      - experiments.py
      - simulator.py
      - __init__.py
      - README.md
      - ipopt_bin
        coin-license.txt
        README.md
      - doc
        gams_2013.txt
    - delayed_impact
      - experiments.py
      - simulator.py
      - data
        totals.csv
        transrisk_performance_by_race_ssa.csv
        transrisk_cdf_by_race_ssa.csv
      - __init__.py
      - README.md
      - fico.py
    - __init__.py
    - hiv
      - environments.py
      - experiments.py
      - simulator.py
      - __init__.py
      - README.md
    - lotka_volterra
      - experiments.py
      - simulator.py
      - __init__.py
      - README.md
    - world2
      - experiments.py
      - simulator.py
      - tables.py
      - __init__.py
      - README.md
    - zika
      - environments.py
      - experiments.py
      - simulator.py
      - __init__.py
      - README.md
    - incarceration
      - experiments.py
      - simulator.py
      - __init__.py
      - README.md
    - schelling
      - experiments.py
      - simulator.py
      - model.py
      - __init__.py
      - README.md
    - credit
      - environments.py
      - simulator.py
      - dataloader.py
      - __init__.py
      - README.md
  - traceable_numpy
    - __init__.py
- examples
  - causal_inference
    - unobserved_confounding.ipynb
    - causal_discovery.ipynb
    - regression_discontinuity.ipynb
    - heterogeneous_treatment_effects.ipynb
    - interference.ipynb
    - graphical_methods.ipynb
    - mediation_analysis.ipynb
    - sensitivity_analysis.ipynb
    - instrumental_variables.ipynb
    - scripts
      - causal_search.py
      - mediator_utils.py
      - heterogeneous_utils.py
    - assets
      - lotka_volterra_graph.gml
      - mediators.svg
      - sensitivity_plots
  - reinforcement_learning
    - world3_simulator.ipynb
    - zika_simulator.ipynb
    - scripts
      - utils.py
  - walkthrough
    - walkthrough.ipynb
  - dynamic_decisions
    - scripts
      - utils.py
    - delayed_impact.ipynb
- .pre-commit-config.yaml
- setup.py
- .travis.yml
- README.md
- tests
  - gym_tests
    - test_envs.py
    - spec_list.py
    - test_registration.py
  - test_world2.py
  - test_causal_graphs.py
  - test_opioid.py
  - gamsBaseline.csv
  - test_dice.py
  - test_dynamics.py
  - test_delayed_impact.py
  - test_world3.py
  - test_framework.py
  - __init__.py
  - benchmark.py
  - test_algorithms.py
  - test_simulators.py
- pytest.ini
- requirements.txt
- lint.sh
- .gitignore
- docs
  - Makefile
  - source
    - estimators.rst
    - designing_experiments.rst
    - causal_graph_discovery.rst
    - causal_inference.rst
    - why.rst
    - usage.rst
    - api.rst
    - references.bib
    - examples.rst
    - treatment_effects.rst
    - develop.rst
    - reinforcement_learning.rst
    - quickstart.rst
    - index.rst
    - conf.py
    - simulators.rst
    - simulator_configs
      - civil_violence.rst
      - credit.rst
      - schelling.rst
      - lotka_volterra.rst
      - world2.rst
      - lalonde.rst
      - incarceration.rst
      - hiv.rst
      - zika.rst
      - opioid.rst
      - dice.rst
      - delayed_impact.rst
      - world3.rst
    - _static
      - WhyNot_fullcolor.svg
      - dynamics.svg
    - installation.rst
  - make.bat
  - .gitignore
- LICENSE.txt
- MANIFEST.in

"""Utility functions for mediation experiment."""
import matplotlib.pylab as plt
import numpy as np


def error_bar_plot(experiment_data, results, title="", ylabel=""):

    true_effect = experiment_data.true_effects.mean()
    estimators = list(results.keys())

    x = list(estimators)
    y = [results[estimator].ate for estimator in estimators]

    cis = [
        np.array(results[estimator].ci) - results[estimator].ate
        if results[estimator].ci is not None
        else [0, 0]
        for estimator in estimators
    ]
    err = [[abs(ci[0]) for ci in cis], [abs(ci[1]) for ci in cis]]

    plt.figure(figsize=(12, 5))
    (_, caps, _) = plt.errorbar(x, y, yerr=err, fmt="o", markersize=8, capsize=5)
    for cap in caps:
        cap.set_markeredgewidth(2)
    plt.plot(x, [true_effect] * len(x), label="True Effect")
    plt.legend(fontsize=12, loc="lower right")
    plt.ylabel(ylabel)
    plt.title(title)