{ "cells": [ { "cell_type": "markdown", "id": "4bbf3e5d", "metadata": {}, "source": [ "# Effective sample size (Python)\n", "\n", ":::{include} ess.md\n", ":start-after: intro-start\n", ":end-before: intro-end\n", ":::" ] }, { "cell_type": "code", "execution_count": 1, "id": "b8e1b6f1", "metadata": {}, "outputs": [ { "data": { "text/html": [ "
\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "
<xarray.Dataset>\n",
       "Dimensions:  (school: 8)\n",
       "Coordinates:\n",
       "  * school   (school) object 'Choate' 'Deerfield' ... "St. Paul's" 'Mt. Hermon'\n",
       "Data variables:\n",
       "    mu       float64 2.354e+03\n",
       "    theta_t  (school) float64 2.215e+03 3.159e+03 ... 2.678e+03 2.522e+03\n",
       "    tau      float64 1.268e+03\n",
       "    theta    (school) float64 2.298e+03 2.434e+03 ... 2.174e+03 2.278e+03
" ], "text/plain": [ "\n", "Dimensions: (school: 8)\n", "Coordinates:\n", " * school (school) object 'Choate' 'Deerfield' ... \"St. Paul's\" 'Mt. Hermon'\n", "Data variables:\n", " mu float64 2.354e+03\n", " theta_t (school) float64 2.215e+03 3.159e+03 ... 2.678e+03 2.522e+03\n", " tau float64 1.268e+03\n", " theta (school) float64 2.298e+03 2.434e+03 ... 2.174e+03 2.278e+03" ] }, "execution_count": 1, "metadata": {}, "output_type": "execute_result" } ], "source": [ "import arviz as az\n", "idata = az.load_arviz_data(\"non_centered_eight\")\n", "az.ess(idata)" ] }, { "cell_type": "markdown", "id": "96c641f2", "metadata": {}, "source": [ ":::{include} ess.md\n", ":start-after: tail-note-start\n", ":end-before: tail-note-end\n", ":::" ] }, { "cell_type": "code", "execution_count": 2, "id": "5a95d44a", "metadata": {}, "outputs": [ { "data": { "text/html": [ "
\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "
<xarray.Dataset>\n",
       "Dimensions:  (school: 8)\n",
       "Coordinates:\n",
       "  * school   (school) object 'Choate' 'Deerfield' ... "St. Paul's" 'Mt. Hermon'\n",
       "Data variables:\n",
       "    mu       float64 1.401e+03\n",
       "    theta_t  (school) float64 1.45e+03 1.514e+03 ... 1.207e+03 1.589e+03\n",
       "    tau      float64 900.0\n",
       "    theta    (school) float64 1.445e+03 1.506e+03 ... 1.433e+03 1.418e+03
" ], "text/plain": [ "\n", "Dimensions: (school: 8)\n", "Coordinates:\n", " * school (school) object 'Choate' 'Deerfield' ... \"St. Paul's\" 'Mt. Hermon'\n", "Data variables:\n", " mu float64 1.401e+03\n", " theta_t (school) float64 1.45e+03 1.514e+03 ... 1.207e+03 1.589e+03\n", " tau float64 900.0\n", " theta (school) float64 1.445e+03 1.506e+03 ... 1.433e+03 1.418e+03" ] }, "execution_count": 2, "metadata": {}, "output_type": "execute_result" } ], "source": [ "az.ess(idata, method=\"tail\")" ] }, { "cell_type": "markdown", "id": "9cfa4410", "metadata": {}, "source": [ ":::{include} ess.md\n", ":start-after: conclusion-start\n", ":end-before: conclusion-end\n", ":::" ] }, { "cell_type": "code", "execution_count": null, "id": "aeb61ed1", "metadata": {}, "outputs": [], "source": [] } ], "metadata": { "kernelspec": { "display_name": "Python 3 (ipykernel)", "language": "python", "name": "python3" }, "language_info": { "codemirror_mode": { "name": "ipython", "version": 3 }, "file_extension": ".py", "mimetype": "text/x-python", "name": "python", "nbconvert_exporter": "python", "pygments_lexer": "ipython3", "version": "3.9.7" } }, "nbformat": 4, "nbformat_minor": 5 }