{ "cells": [ { "cell_type": "markdown", "metadata": {}, "source": [ "(working_with_InferenceData)=\n", "\n", "# Working with InferenceData\n", "\n", "Here we present a collection of common manipulations you can use while working with `InferenceData`." ] }, { "cell_type": "code", "execution_count": 3, "metadata": {}, "outputs": [], "source": [ "import arviz as az\n", "import numpy as np\n", "import xarray as xr\n", "\n", "xr.set_options(display_expand_data=False, display_expand_attrs=False);" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "`display_expand_data=False` makes the default view for {class}`xarray.DataArray` fold the data values to a single line. To explore the values, click on the {fas}`database` icon on the left of the view, right under the `xarray.DataArray` text. It has no effect on `Dataset` objects that already default to folded views.\n", "\n", "`display_expand_attrs=False` folds the attributes in both `DataArray` and `Dataset` objects to keep the views shorter. In this page we print DataArrays and Datasets several times and they always have the same attributes." ] }, { "cell_type": "code", "execution_count": 4, "metadata": {}, "outputs": [ { "data": { "text/html": [ "\n", "
<xarray.Dataset>\n", "Dimensions: (chain: 4, draw: 500, school: 8)\n", "Coordinates:\n", " * chain (chain) int64 0 1 2 3\n", " * draw (draw) int64 0 1 2 3 4 5 6 7 8 ... 492 493 494 495 496 497 498 499\n", " * school (school) <U16 'Choate' 'Deerfield' ... "St. Paul's" 'Mt. Hermon'\n", "Data variables:\n", " mu (chain, draw) float64 ...\n", " theta (chain, draw, school) float64 ...\n", " tau (chain, draw) float64 ...\n", "Attributes: (6)
<xarray.Dataset>\n", "Dimensions: (chain: 4, draw: 500, school: 8)\n", "Coordinates:\n", " * chain (chain) int64 0 1 2 3\n", " * draw (draw) int64 0 1 2 3 4 5 6 7 8 ... 492 493 494 495 496 497 498 499\n", " * school (school) <U16 'Choate' 'Deerfield' ... "St. Paul's" 'Mt. Hermon'\n", "Data variables:\n", " obs (chain, draw, school) float64 ...\n", "Attributes: (4)
<xarray.Dataset>\n", "Dimensions: (chain: 4, draw: 500, school: 8)\n", "Coordinates:\n", " * chain (chain) int64 0 1 2 3\n", " * draw (draw) int64 0 1 2 3 4 5 6 7 8 ... 492 493 494 495 496 497 498 499\n", " * school (school) <U16 'Choate' 'Deerfield' ... "St. Paul's" 'Mt. Hermon'\n", "Data variables:\n", " obs (chain, draw, school) float64 ...\n", "Attributes: (4)
<xarray.Dataset>\n", "Dimensions: (chain: 4, draw: 500)\n", "Coordinates:\n", " * chain (chain) int64 0 1 2 3\n", " * draw (draw) int64 0 1 2 3 4 5 6 ... 494 495 496 497 498 499\n", "Data variables: (12/16)\n", " max_energy_error (chain, draw) float64 ...\n", " energy_error (chain, draw) float64 ...\n", " lp (chain, draw) float64 ...\n", " index_in_trajectory (chain, draw) int64 ...\n", " acceptance_rate (chain, draw) float64 ...\n", " diverging (chain, draw) bool ...\n", " ... ...\n", " smallest_eigval (chain, draw) float64 ...\n", " step_size_bar (chain, draw) float64 ...\n", " step_size (chain, draw) float64 ...\n", " energy (chain, draw) float64 ...\n", " tree_depth (chain, draw) int64 ...\n", " perf_counter_diff (chain, draw) float64 ...\n", "Attributes: (6)
<xarray.Dataset>\n", "Dimensions: (chain: 1, draw: 500, school: 8)\n", "Coordinates:\n", " * chain (chain) int64 0\n", " * draw (draw) int64 0 1 2 3 4 5 6 7 8 ... 492 493 494 495 496 497 498 499\n", " * school (school) <U16 'Choate' 'Deerfield' ... "St. Paul's" 'Mt. Hermon'\n", "Data variables:\n", " tau (chain, draw) float64 ...\n", " theta (chain, draw, school) float64 ...\n", " mu (chain, draw) float64 ...\n", "Attributes: (4)
<xarray.Dataset>\n", "Dimensions: (chain: 1, draw: 500, school: 8)\n", "Coordinates:\n", " * chain (chain) int64 0\n", " * draw (draw) int64 0 1 2 3 4 5 6 7 8 ... 492 493 494 495 496 497 498 499\n", " * school (school) <U16 'Choate' 'Deerfield' ... "St. Paul's" 'Mt. Hermon'\n", "Data variables:\n", " obs (chain, draw, school) float64 ...\n", "Attributes: (4)
<xarray.Dataset>\n", "Dimensions: (school: 8)\n", "Coordinates:\n", " * school (school) <U16 'Choate' 'Deerfield' ... "St. Paul's" 'Mt. Hermon'\n", "Data variables:\n", " obs (school) float64 ...\n", "Attributes: (4)
<xarray.Dataset>\n", "Dimensions: (school: 8)\n", "Coordinates:\n", " * school (school) <U16 'Choate' 'Deerfield' ... "St. Paul's" 'Mt. Hermon'\n", "Data variables:\n", " scores (school) float64 ...\n", "Attributes: (4)
<xarray.Dataset>\n", "Dimensions: (chain: 4, draw: 500, school: 8)\n", "Coordinates:\n", " * chain (chain) int64 0 1 2 3\n", " * draw (draw) int64 0 1 2 3 4 5 6 7 8 ... 492 493 494 495 496 497 498 499\n", " * school (school) <U16 'Choate' 'Deerfield' ... "St. Paul's" 'Mt. Hermon'\n", "Data variables:\n", " mu (chain, draw) float64 ...\n", " theta (chain, draw, school) float64 ...\n", " tau (chain, draw) float64 ...\n", "Attributes: (6)
<xarray.Dataset>\n", "Dimensions: (chain: 4, draw: 500, school: 8)\n", "Coordinates:\n", " * chain (chain) int64 0 1 2 3\n", " * draw (draw) int64 0 1 2 3 4 5 6 7 8 ... 492 493 494 495 496 497 498 499\n", " * school (school) <U16 'Choate' 'Deerfield' ... "St. Paul's" 'Mt. Hermon'\n", "Data variables:\n", " mu (chain, draw) float64 ...\n", " theta (chain, draw, school) float64 ...\n", " tau (chain, draw) float64 4.726 3.909 4.844 1.857 ... 2.741 2.932 4.461\n", " log_tau (chain, draw) float64 1.553 1.363 1.578 ... 1.008 1.076 1.495\n", "Attributes: (6)" ], "text/plain": [ "
<xarray.Dataset>\n", "Dimensions: (sample: 2000, school: 8)\n", "Coordinates:\n", " * school (school) <U16 'Choate' 'Deerfield' ... "St. Paul's" 'Mt. Hermon'\n", " * sample (sample) object MultiIndex\n", " * chain (sample) int64 0 0 0 0 0 0 0 0 0 0 0 0 ... 3 3 3 3 3 3 3 3 3 3 3 3\n", " * draw (sample) int64 0 1 2 3 4 5 6 7 ... 492 493 494 495 496 497 498 499\n", "Data variables:\n", " mu (sample) float64 7.872 3.385 9.1 7.304 ... 1.859 1.767 3.486 3.404\n", " theta (school, sample) float64 12.32 11.29 5.709 ... -2.623 8.452 1.295\n", " tau (sample) float64 4.726 3.909 4.844 1.857 ... 2.741 2.932 4.461\n", " log_tau (sample) float64 1.553 1.363 1.578 0.6188 ... 1.008 1.076 1.495\n", "Attributes: (6)" ], "text/plain": [ "
<xarray.Dataset>\n", "Dimensions: (sample: 100, school: 8)\n", "Coordinates:\n", " * school (school) <U16 'Choate' 'Deerfield' ... "St. Paul's" 'Mt. Hermon'\n", " * sample (sample) object MultiIndex\n", " * chain (sample) int64 3 1 1 3 2 3 3 3 3 1 0 0 ... 2 0 2 2 3 2 2 3 2 0 3 2\n", " * draw (sample) int64 214 202 176 487 371 27 ... 207 411 170 102 70 385\n", "Data variables:\n", " mu (sample) float64 10.14 0.2007 12.31 0.4654 ... -5.88 12.93 4.57\n", " theta (school, sample) float64 10.01 24.78 14.16 ... -4.128 14.04 3.417\n", " tau (sample) float64 3.008 19.69 1.974 3.611 ... 2.457 1.675 2.081\n", " log_tau (sample) float64 1.101 2.98 0.6802 1.284 ... 0.8991 0.5161 0.733\n", "Attributes: (6)" ], "text/plain": [ "
<xarray.DataArray 'school' (school: 8)>\n", "'Choate' 'Deerfield' 'Phillips Andover' ... "St. Paul's" 'Mt. Hermon'\n", "Coordinates:\n", " * school (school) <U16 'Choate' 'Deerfield' ... "St. Paul's" 'Mt. Hermon'" ], "text/plain": [ "
<xarray.Dataset>\n", "Dimensions: (chain: 2, draw: 500, school: 8)\n", "Coordinates:\n", " * chain (chain) int64 0 2\n", " * draw (draw) int64 0 1 2 3 4 5 6 7 8 ... 492 493 494 495 496 497 498 499\n", " * school (school) <U16 'Choate' 'Deerfield' ... "St. Paul's" 'Mt. Hermon'\n", "Data variables:\n", " mu (chain, draw) float64 7.872 3.385 9.1 7.304 ... 2.871 4.096 1.776\n", " theta (chain, draw, school) float64 12.32 9.905 14.95 ... 2.363 -2.968\n", " tau (chain, draw) float64 4.726 3.909 4.844 1.857 ... 4.09 2.72 1.917\n", " log_tau (chain, draw) float64 1.553 1.363 1.578 ... 1.408 1.001 0.6508\n", "Attributes: (6)
<xarray.Dataset>\n", "Dimensions: (chain: 2, draw: 500, school: 8)\n", "Coordinates:\n", " * chain (chain) int64 0 2\n", " * draw (draw) int64 0 1 2 3 4 5 6 7 8 ... 492 493 494 495 496 497 498 499\n", " * school (school) <U16 'Choate' 'Deerfield' ... "St. Paul's" 'Mt. Hermon'\n", "Data variables:\n", " obs (chain, draw, school) float64 ...\n", "Attributes: (4)
<xarray.Dataset>\n", "Dimensions: (chain: 2, draw: 500, school: 8)\n", "Coordinates:\n", " * chain (chain) int64 0 2\n", " * draw (draw) int64 0 1 2 3 4 5 6 7 8 ... 492 493 494 495 496 497 498 499\n", " * school (school) <U16 'Choate' 'Deerfield' ... "St. Paul's" 'Mt. Hermon'\n", "Data variables:\n", " obs (chain, draw, school) float64 ...\n", "Attributes: (4)
<xarray.Dataset>\n", "Dimensions: (chain: 2, draw: 500)\n", "Coordinates:\n", " * chain (chain) int64 0 2\n", " * draw (draw) int64 0 1 2 3 4 5 6 ... 494 495 496 497 498 499\n", "Data variables: (12/16)\n", " max_energy_error (chain, draw) float64 ...\n", " energy_error (chain, draw) float64 ...\n", " lp (chain, draw) float64 ...\n", " index_in_trajectory (chain, draw) int64 ...\n", " acceptance_rate (chain, draw) float64 ...\n", " diverging (chain, draw) bool ...\n", " ... ...\n", " smallest_eigval (chain, draw) float64 ...\n", " step_size_bar (chain, draw) float64 ...\n", " step_size (chain, draw) float64 ...\n", " energy (chain, draw) float64 ...\n", " tree_depth (chain, draw) int64 ...\n", " perf_counter_diff (chain, draw) float64 ...\n", "Attributes: (6)
<xarray.Dataset>\n", "Dimensions: (chain: 1, draw: 500, school: 8)\n", "Coordinates:\n", " * chain (chain) int64 0\n", " * draw (draw) int64 0 1 2 3 4 5 6 7 8 ... 492 493 494 495 496 497 498 499\n", " * school (school) <U16 'Choate' 'Deerfield' ... "St. Paul's" 'Mt. Hermon'\n", "Data variables:\n", " tau (chain, draw) float64 ...\n", " theta (chain, draw, school) float64 ...\n", " mu (chain, draw) float64 ...\n", "Attributes: (4)
<xarray.Dataset>\n", "Dimensions: (chain: 1, draw: 500, school: 8)\n", "Coordinates:\n", " * chain (chain) int64 0\n", " * draw (draw) int64 0 1 2 3 4 5 6 7 8 ... 492 493 494 495 496 497 498 499\n", " * school (school) <U16 'Choate' 'Deerfield' ... "St. Paul's" 'Mt. Hermon'\n", "Data variables:\n", " obs (chain, draw, school) float64 ...\n", "Attributes: (4)
<xarray.Dataset>\n", "Dimensions: (school: 8)\n", "Coordinates:\n", " * school (school) <U16 'Choate' 'Deerfield' ... "St. Paul's" 'Mt. Hermon'\n", "Data variables:\n", " obs (school) float64 ...\n", "Attributes: (4)
<xarray.Dataset>\n", "Dimensions: (school: 8)\n", "Coordinates:\n", " * school (school) <U16 'Choate' 'Deerfield' ... "St. Paul's" 'Mt. Hermon'\n", "Data variables:\n", " scores (school) float64 ...\n", "Attributes: (4)
<xarray.Dataset>\n", "Dimensions: (chain: 4, draw: 400, school: 8)\n", "Coordinates:\n", " * chain (chain) int64 0 1 2 3\n", " * draw (draw) int64 100 101 102 103 104 105 ... 494 495 496 497 498 499\n", " * school (school) <U16 'Choate' 'Deerfield' ... "St. Paul's" 'Mt. Hermon'\n", "Data variables:\n", " mu (chain, draw) float64 11.7 8.118 -5.88 -7.149 ... 1.767 3.486 3.404\n", " theta (chain, draw, school) float64 14.23 9.72 9.195 ... 6.762 1.295\n", " tau (chain, draw) float64 4.289 2.765 2.457 1.719 ... 2.741 2.932 4.461\n", " log_tau (chain, draw) float64 1.456 1.017 0.8991 ... 1.008 1.076 1.495\n", "Attributes: (6)
<xarray.Dataset>\n", "Dimensions: (chain: 4, draw: 400, school: 8)\n", "Coordinates:\n", " * chain (chain) int64 0 1 2 3\n", " * draw (draw) int64 100 101 102 103 104 105 ... 494 495 496 497 498 499\n", " * school (school) <U16 'Choate' 'Deerfield' ... "St. Paul's" 'Mt. Hermon'\n", "Data variables:\n", " obs (chain, draw, school) float64 ...\n", "Attributes: (4)
<xarray.Dataset>\n", "Dimensions: (chain: 4, draw: 400, school: 8)\n", "Coordinates:\n", " * chain (chain) int64 0 1 2 3\n", " * draw (draw) int64 100 101 102 103 104 105 ... 494 495 496 497 498 499\n", " * school (school) <U16 'Choate' 'Deerfield' ... "St. Paul's" 'Mt. Hermon'\n", "Data variables:\n", " obs (chain, draw, school) float64 ...\n", "Attributes: (4)
<xarray.Dataset>\n", "Dimensions: (chain: 4, draw: 400)\n", "Coordinates:\n", " * chain (chain) int64 0 1 2 3\n", " * draw (draw) int64 100 101 102 103 104 ... 496 497 498 499\n", "Data variables: (12/16)\n", " max_energy_error (chain, draw) float64 ...\n", " energy_error (chain, draw) float64 ...\n", " lp (chain, draw) float64 ...\n", " index_in_trajectory (chain, draw) int64 ...\n", " acceptance_rate (chain, draw) float64 ...\n", " diverging (chain, draw) bool ...\n", " ... ...\n", " smallest_eigval (chain, draw) float64 ...\n", " step_size_bar (chain, draw) float64 ...\n", " step_size (chain, draw) float64 ...\n", " energy (chain, draw) float64 ...\n", " tree_depth (chain, draw) int64 ...\n", " perf_counter_diff (chain, draw) float64 ...\n", "Attributes: (6)
<xarray.Dataset>\n", "Dimensions: (chain: 1, draw: 400, school: 8)\n", "Coordinates:\n", " * chain (chain) int64 0\n", " * draw (draw) int64 100 101 102 103 104 105 ... 494 495 496 497 498 499\n", " * school (school) <U16 'Choate' 'Deerfield' ... "St. Paul's" 'Mt. Hermon'\n", "Data variables:\n", " tau (chain, draw) float64 ...\n", " theta (chain, draw, school) float64 ...\n", " mu (chain, draw) float64 ...\n", "Attributes: (4)
<xarray.Dataset>\n", "Dimensions: (chain: 1, draw: 400, school: 8)\n", "Coordinates:\n", " * chain (chain) int64 0\n", " * draw (draw) int64 100 101 102 103 104 105 ... 494 495 496 497 498 499\n", " * school (school) <U16 'Choate' 'Deerfield' ... "St. Paul's" 'Mt. Hermon'\n", "Data variables:\n", " obs (chain, draw, school) float64 ...\n", "Attributes: (4)
<xarray.Dataset>\n", "Dimensions: (school: 8)\n", "Coordinates:\n", " * school (school) <U16 'Choate' 'Deerfield' ... "St. Paul's" 'Mt. Hermon'\n", "Data variables:\n", " obs (school) float64 ...\n", "Attributes: (4)
<xarray.Dataset>\n", "Dimensions: (school: 8)\n", "Coordinates:\n", " * school (school) <U16 'Choate' 'Deerfield' ... "St. Paul's" 'Mt. Hermon'\n", "Data variables:\n", " scores (school) float64 ...\n", "Attributes: (4)
<xarray.Dataset>\n", "Dimensions: (chain: 4, draw: 400, school: 8)\n", "Coordinates:\n", " * chain (chain) int64 0 1 2 3\n", " * draw (draw) int64 100 101 102 103 104 105 ... 494 495 496 497 498 499\n", " * school (school) <U16 'Choate' 'Deerfield' ... "St. Paul's" 'Mt. Hermon'\n", "Data variables:\n", " mu (chain, draw) float64 11.7 8.118 -5.88 -7.149 ... 1.767 3.486 3.404\n", " theta (chain, draw, school) float64 14.23 9.72 9.195 ... 6.762 1.295\n", " tau (chain, draw) float64 4.289 2.765 2.457 1.719 ... 2.741 2.932 4.461\n", " log_tau (chain, draw) float64 1.456 1.017 0.8991 ... 1.008 1.076 1.495\n", "Attributes: (6)
<xarray.Dataset>\n", "Dimensions: (chain: 4, draw: 500, school: 8)\n", "Coordinates:\n", " * chain (chain) int64 0 1 2 3\n", " * draw (draw) int64 0 1 2 3 4 5 6 7 8 ... 492 493 494 495 496 497 498 499\n", " * school (school) <U16 'Choate' 'Deerfield' ... "St. Paul's" 'Mt. Hermon'\n", "Data variables:\n", " obs (chain, draw, school) float64 ...\n", "Attributes: (4)
<xarray.Dataset>\n", "Dimensions: (chain: 4, draw: 500, school: 8)\n", "Coordinates:\n", " * chain (chain) int64 0 1 2 3\n", " * draw (draw) int64 0 1 2 3 4 5 6 7 8 ... 492 493 494 495 496 497 498 499\n", " * school (school) <U16 'Choate' 'Deerfield' ... "St. Paul's" 'Mt. Hermon'\n", "Data variables:\n", " obs (chain, draw, school) float64 ...\n", "Attributes: (4)
<xarray.Dataset>\n", "Dimensions: (chain: 4, draw: 500)\n", "Coordinates:\n", " * chain (chain) int64 0 1 2 3\n", " * draw (draw) int64 0 1 2 3 4 5 6 ... 494 495 496 497 498 499\n", "Data variables: (12/16)\n", " max_energy_error (chain, draw) float64 ...\n", " energy_error (chain, draw) float64 ...\n", " lp (chain, draw) float64 ...\n", " index_in_trajectory (chain, draw) int64 ...\n", " acceptance_rate (chain, draw) float64 ...\n", " diverging (chain, draw) bool ...\n", " ... ...\n", " smallest_eigval (chain, draw) float64 ...\n", " step_size_bar (chain, draw) float64 ...\n", " step_size (chain, draw) float64 ...\n", " energy (chain, draw) float64 ...\n", " tree_depth (chain, draw) int64 ...\n", " perf_counter_diff (chain, draw) float64 ...\n", "Attributes: (6)
<xarray.Dataset>\n", "Dimensions: (chain: 1, draw: 500, school: 8)\n", "Coordinates:\n", " * chain (chain) int64 0\n", " * draw (draw) int64 0 1 2 3 4 5 6 7 8 ... 492 493 494 495 496 497 498 499\n", " * school (school) <U16 'Choate' 'Deerfield' ... "St. Paul's" 'Mt. Hermon'\n", "Data variables:\n", " tau (chain, draw) float64 ...\n", " theta (chain, draw, school) float64 ...\n", " mu (chain, draw) float64 ...\n", "Attributes: (4)
<xarray.Dataset>\n", "Dimensions: (chain: 1, draw: 500, school: 8)\n", "Coordinates:\n", " * chain (chain) int64 0\n", " * draw (draw) int64 0 1 2 3 4 5 6 7 8 ... 492 493 494 495 496 497 498 499\n", " * school (school) <U16 'Choate' 'Deerfield' ... "St. Paul's" 'Mt. Hermon'\n", "Data variables:\n", " obs (chain, draw, school) float64 ...\n", "Attributes: (4)
<xarray.Dataset>\n", "Dimensions: (school: 8)\n", "Coordinates:\n", " * school (school) <U16 'Choate' 'Deerfield' ... "St. Paul's" 'Mt. Hermon'\n", "Data variables:\n", " obs (school) float64 ...\n", "Attributes: (4)
<xarray.Dataset>\n", "Dimensions: (school: 8)\n", "Coordinates:\n", " * school (school) <U16 'Choate' 'Deerfield' ... "St. Paul's" 'Mt. Hermon'\n", "Data variables:\n", " scores (school) float64 ...\n", "Attributes: (4)
<xarray.Dataset>\n", "Dimensions: ()\n", "Data variables:\n", " mu float64 4.486\n", " theta float64 4.912\n", " tau float64 4.124\n", " log_tau float64 1.173
<xarray.Dataset>\n", "Dimensions: (school: 8)\n", "Coordinates:\n", " * school (school) <U16 'Choate' 'Deerfield' ... "St. Paul's" 'Mt. Hermon'\n", "Data variables:\n", " mu float64 4.486\n", " theta (school) float64 6.46 5.028 3.938 4.872 3.667 3.975 6.581 4.772\n", " tau float64 4.124\n", " log_tau float64 1.173
<xarray.Dataset>\n", "Dimensions: (chain: 4, draw: 500, school: 8, school_bis: 8)\n", "Coordinates:\n", " * chain (chain) int64 0 1 2 3\n", " * draw (draw) int64 0 1 2 3 4 5 6 ... 494 495 496 497 498 499\n", " * school (school) <U16 'Choate' 'Deerfield' ... 'Mt. Hermon'\n", " * school_bis (school_bis) <U16 'Choate' 'Deerfield' ... 'Mt. Hermon'\n", "Data variables:\n", " mu (chain, draw) float64 7.872 3.385 9.1 ... 3.486 3.404\n", " theta (chain, draw, school) float64 12.32 9.905 ... 6.762 1.295\n", " tau (chain, draw) float64 4.726 3.909 4.844 ... 2.932 4.461\n", " log_tau (chain, draw) float64 1.553 1.363 1.578 ... 1.076 1.495\n", " mlogtau (chain, draw) float64 nan nan nan ... 1.494 1.496 1.511\n", " theta_school_diff (chain, draw, school, school_bis) float64 0.0 ... 0.0\n", "Attributes: (6)" ], "text/plain": [ "
<xarray.DataArray 'theta_school_diff' (chain: 4, draw: 500)>\n", "2.415 2.156 -0.04943 1.228 3.384 9.662 ... -1.656 -0.4021 1.524 -3.372 -6.305\n", "Coordinates:\n", " * chain (chain) int64 0 1 2 3\n", " * draw (draw) int64 0 1 2 3 4 5 6 7 ... 492 493 494 495 496 497 498 499\n", " school <U16 'Choate'\n", " school_bis <U16 'Deerfield'" ], "text/plain": [ "
<xarray.DataArray 'theta_school_diff' (chain: 4, draw: 500,\n", " pairwise_school_diff: 3)>\n", "2.415 -6.741 -1.84 2.156 -3.474 3.784 ... -2.619 6.923 -6.305 1.667 -6.641\n", "Coordinates:\n", " * chain (chain) int64 0 1 2 3\n", " * draw (draw) int64 0 1 2 3 4 5 6 7 ... 492 493 494 495 496 497 498 499\n", " school (pairwise_school_diff) <U16 'Choate' 'Hotchkiss' 'Mt. Hermon'\n", " school_bis (pairwise_school_diff) <U16 'Deerfield' 'Choate' 'Lawrenceville'\n", "Dimensions without coordinates: pairwise_school_diff" ], "text/plain": [ "
<xarray.DataArray 'theta_school_diff' (chain: 4, draw: 500, school: 3,\n", " school_bis: 3)>\n", "2.415 0.0 -4.581 -4.326 -6.741 -11.32 ... 1.667 -6.077 -5.203 1.102 -6.641\n", "Coordinates:\n", " * chain (chain) int64 0 1 2 3\n", " * draw (draw) int64 0 1 2 3 4 5 6 7 ... 492 493 494 495 496 497 498 499\n", " * school (school) <U16 'Choate' 'Hotchkiss' 'Mt. Hermon'\n", " * school_bis (school_bis) <U16 'Deerfield' 'Choate' 'Lawrenceville'" ], "text/plain": [ "
<xarray.Dataset>\n", "Dimensions: (chain: 4, draw: 500, school: 8, school_bis: 8)\n", "Coordinates:\n", " * chain (chain) int64 0 1 2 3\n", " * draw (draw) int64 0 1 2 3 4 5 6 ... 494 495 496 497 498 499\n", " * school (school) <U16 'Choate' 'Deerfield' ... 'Mt. Hermon'\n", " * school_bis (school_bis) <U16 'Choate' 'Deerfield' ... 'Mt. Hermon'\n", "Data variables:\n", " mu (chain, draw) float64 7.872 3.385 9.1 ... 3.486 3.404\n", " theta (chain, draw, school) float64 12.32 9.905 ... 6.762 1.295\n", " tau (chain, draw) float64 4.726 3.909 4.844 ... 2.932 4.461\n", " log_tau (chain, draw) float64 1.553 1.363 1.578 ... 1.076 1.495\n", " mlogtau (chain, draw) float64 nan nan nan ... 1.494 1.496 1.511\n", " theta_school_diff (chain, draw, school, school_bis) float64 0.0 ... 0.0\n", "Attributes: (6)
<xarray.Dataset>\n", "Dimensions: (chain: 4, draw: 500, school: 8)\n", "Coordinates:\n", " * chain (chain) int64 0 1 2 3\n", " * draw (draw) int64 0 1 2 3 4 5 6 7 8 ... 492 493 494 495 496 497 498 499\n", " * school (school) <U16 'Choate' 'Deerfield' ... "St. Paul's" 'Mt. Hermon'\n", "Data variables:\n", " obs (chain, draw, school) float64 ...\n", "Attributes: (4)
<xarray.Dataset>\n", "Dimensions: (chain: 4, draw: 500, new_school: 2)\n", "Coordinates:\n", " * chain (chain) int64 0 1 2 3\n", " * draw (draw) int64 0 1 2 3 4 5 6 7 ... 492 493 494 495 496 497 498 499\n", " * new_school (new_school) <U13 'Essex College' 'Moordale'\n", "Data variables:\n", " obs (chain, draw, new_school) float64 2.041 -2.556 ... -0.2822\n", "Attributes: (2)
<xarray.Dataset>\n", "Dimensions: (chain: 4, draw: 500, school: 8)\n", "Coordinates:\n", " * chain (chain) int64 0 1 2 3\n", " * draw (draw) int64 0 1 2 3 4 5 6 7 8 ... 492 493 494 495 496 497 498 499\n", " * school (school) <U16 'Choate' 'Deerfield' ... "St. Paul's" 'Mt. Hermon'\n", "Data variables:\n", " obs (chain, draw, school) float64 ...\n", "Attributes: (4)
<xarray.Dataset>\n", "Dimensions: (chain: 4, draw: 500)\n", "Coordinates:\n", " * chain (chain) int64 0 1 2 3\n", " * draw (draw) int64 0 1 2 3 4 5 6 ... 494 495 496 497 498 499\n", "Data variables: (12/16)\n", " max_energy_error (chain, draw) float64 ...\n", " energy_error (chain, draw) float64 ...\n", " lp (chain, draw) float64 ...\n", " index_in_trajectory (chain, draw) int64 ...\n", " acceptance_rate (chain, draw) float64 ...\n", " diverging (chain, draw) bool ...\n", " ... ...\n", " smallest_eigval (chain, draw) float64 ...\n", " step_size_bar (chain, draw) float64 ...\n", " step_size (chain, draw) float64 ...\n", " energy (chain, draw) float64 ...\n", " tree_depth (chain, draw) int64 ...\n", " perf_counter_diff (chain, draw) float64 ...\n", "Attributes: (6)
<xarray.Dataset>\n", "Dimensions: (chain: 1, draw: 500, school: 8)\n", "Coordinates:\n", " * chain (chain) int64 0\n", " * draw (draw) int64 0 1 2 3 4 5 6 7 8 ... 492 493 494 495 496 497 498 499\n", " * school (school) <U16 'Choate' 'Deerfield' ... "St. Paul's" 'Mt. Hermon'\n", "Data variables:\n", " tau (chain, draw) float64 ...\n", " theta (chain, draw, school) float64 ...\n", " mu (chain, draw) float64 ...\n", "Attributes: (4)
<xarray.Dataset>\n", "Dimensions: (chain: 1, draw: 500, school: 8)\n", "Coordinates:\n", " * chain (chain) int64 0\n", " * draw (draw) int64 0 1 2 3 4 5 6 7 8 ... 492 493 494 495 496 497 498 499\n", " * school (school) <U16 'Choate' 'Deerfield' ... "St. Paul's" 'Mt. Hermon'\n", "Data variables:\n", " obs (chain, draw, school) float64 ...\n", "Attributes: (4)
<xarray.Dataset>\n", "Dimensions: (school: 8)\n", "Coordinates:\n", " * school (school) <U16 'Choate' 'Deerfield' ... "St. Paul's" 'Mt. Hermon'\n", "Data variables:\n", " obs (school) float64 ...\n", "Attributes: (4)
<xarray.Dataset>\n", "Dimensions: (school: 8)\n", "Coordinates:\n", " * school (school) <U16 'Choate' 'Deerfield' ... "St. Paul's" 'Mt. Hermon'\n", "Data variables:\n", " scores (school) float64 ...\n", "Attributes: (4)
<xarray.Dataset>\n", "Dimensions: (draw: 500, school: 8, school_bis: 8)\n", "Coordinates:\n", " * draw (draw) int64 0 1 2 3 4 5 6 ... 494 495 496 497 498 499\n", " * school (school) <U16 'Choate' 'Deerfield' ... 'Mt. Hermon'\n", " * school_bis (school_bis) <U16 'Choate' 'Deerfield' ... 'Mt. Hermon'\n", "Data variables:\n", " mu (draw) float64 5.974 5.096 7.177 ... 3.284 4.739 3.146\n", " theta (draw, school) float64 9.519 5.554 6.118 ... 5.595 3.773\n", " tau (draw) float64 4.068 3.156 3.603 ... 2.725 3.225 2.979\n", " log_tau (draw) float64 1.322 1.118 1.234 ... 0.958 1.035 0.9508\n", " mlogtau (draw) float64 nan nan nan nan ... 0.993 1.002 1.01 1.021\n", " theta_school_diff (draw, school, school_bis) float64 0.0 3.965 ... 0.0
<xarray.Dataset>\n", "Dimensions: (chain: 4, draw: 500, school: 8)\n", "Coordinates:\n", " * chain (chain) int64 0 1 2 3\n", " * draw (draw) int64 0 1 2 3 4 5 6 7 8 ... 492 493 494 495 496 497 498 499\n", " * school (school) <U16 'Choate' 'Deerfield' ... "St. Paul's" 'Mt. Hermon'\n", "Data variables:\n", " obs (chain, draw, school) float64 ...\n", "Attributes: (4)
<xarray.Dataset>\n", "Dimensions: (chain: 4, draw: 500, new_school: 2)\n", "Coordinates:\n", " * chain (chain) int64 0 1 2 3\n", " * draw (draw) int64 0 1 2 3 4 5 6 7 ... 492 493 494 495 496 497 498 499\n", " * new_school (new_school) <U13 'Essex College' 'Moordale'\n", "Data variables:\n", " obs (chain, draw, new_school) float64 2.041 -2.556 ... -0.2822\n", "Attributes: (2)
<xarray.Dataset>\n", "Dimensions: (chain: 4, draw: 500, school: 8)\n", "Coordinates:\n", " * chain (chain) int64 0 1 2 3\n", " * draw (draw) int64 0 1 2 3 4 5 6 7 8 ... 492 493 494 495 496 497 498 499\n", " * school (school) <U16 'Choate' 'Deerfield' ... "St. Paul's" 'Mt. Hermon'\n", "Data variables:\n", " obs (chain, draw, school) float64 ...\n", "Attributes: (4)
<xarray.Dataset>\n", "Dimensions: (chain: 4, draw: 500)\n", "Coordinates:\n", " * chain (chain) int64 0 1 2 3\n", " * draw (draw) int64 0 1 2 3 4 5 6 ... 494 495 496 497 498 499\n", "Data variables: (12/16)\n", " max_energy_error (chain, draw) float64 ...\n", " energy_error (chain, draw) float64 ...\n", " lp (chain, draw) float64 ...\n", " index_in_trajectory (chain, draw) int64 ...\n", " acceptance_rate (chain, draw) float64 ...\n", " diverging (chain, draw) bool ...\n", " ... ...\n", " smallest_eigval (chain, draw) float64 ...\n", " step_size_bar (chain, draw) float64 ...\n", " step_size (chain, draw) float64 ...\n", " energy (chain, draw) float64 ...\n", " tree_depth (chain, draw) int64 ...\n", " perf_counter_diff (chain, draw) float64 ...\n", "Attributes: (6)
<xarray.Dataset>\n", "Dimensions: (draw: 500, school: 8)\n", "Coordinates:\n", " * draw (draw) int64 0 1 2 3 4 5 6 7 8 ... 492 493 494 495 496 497 498 499\n", " * school (school) <U16 'Choate' 'Deerfield' ... "St. Paul's" 'Mt. Hermon'\n", "Data variables:\n", " tau (draw) float64 1.941 3.388 4.208 5.687 ... 0.8353 0.06893 2.145\n", " theta (draw, school) float64 4.866 4.59 -0.7404 ... 3.33 -2.031 6.045\n", " mu (draw) float64 3.903 3.915 -1.751 2.595 ... -2.294 0.7908 2.869
<xarray.Dataset>\n", "Dimensions: (chain: 1, draw: 500, school: 8)\n", "Coordinates:\n", " * chain (chain) int64 0\n", " * draw (draw) int64 0 1 2 3 4 5 6 7 8 ... 492 493 494 495 496 497 498 499\n", " * school (school) <U16 'Choate' 'Deerfield' ... "St. Paul's" 'Mt. Hermon'\n", "Data variables:\n", " obs (chain, draw, school) float64 ...\n", "Attributes: (4)
<xarray.Dataset>\n", "Dimensions: (school: 8)\n", "Coordinates:\n", " * school (school) <U16 'Choate' 'Deerfield' ... "St. Paul's" 'Mt. Hermon'\n", "Data variables:\n", " obs (school) float64 ...\n", "Attributes: (4)
<xarray.Dataset>\n", "Dimensions: (school: 8)\n", "Coordinates:\n", " * school (school) <U16 'Choate' 'Deerfield' ... "St. Paul's" 'Mt. Hermon'\n", "Data variables:\n", " scores (school) float64 ...\n", "Attributes: (4)
<xarray.Dataset>\n", "Dimensions: (draw: 500, school: 8, school_bis: 8)\n", "Coordinates:\n", " * draw (draw) int64 0 1 2 3 4 5 6 ... 494 495 496 497 498 499\n", " * school (school) <U16 'Choate' 'Deerfield' ... 'Mt. Hermon'\n", " * school_bis (school_bis) <U16 'Choate' 'Deerfield' ... 'Mt. Hermon'\n", "Data variables:\n", " mu (draw) float64 8.974 8.096 10.18 ... 6.284 7.739 6.146\n", " theta (draw, school) float64 12.52 8.554 9.118 ... 8.595 6.773\n", " tau (draw) float64 7.068 6.156 6.603 ... 5.725 6.225 5.979\n", " log_tau (draw) float64 4.322 4.118 4.234 ... 3.958 4.035 3.951\n", " mlogtau (draw) float64 nan nan nan nan ... 3.993 4.002 4.01 4.021\n", " theta_school_diff (draw, school, school_bis) float64 3.0 6.965 ... 3.0
<xarray.Dataset>\n", "Dimensions: (chain: 4, draw: 500, school: 8)\n", "Coordinates:\n", " * chain (chain) int64 0 1 2 3\n", " * draw (draw) int64 0 1 2 3 4 5 6 7 8 ... 492 493 494 495 496 497 498 499\n", " * school (school) <U16 'Choate' 'Deerfield' ... "St. Paul's" 'Mt. Hermon'\n", "Data variables:\n", " obs (chain, draw, school) float64 ...\n", "Attributes: (4)
<xarray.Dataset>\n", "Dimensions: (chain: 4, draw: 500, new_school: 2)\n", "Coordinates:\n", " * chain (chain) int64 0 1 2 3\n", " * draw (draw) int64 0 1 2 3 4 5 6 7 ... 492 493 494 495 496 497 498 499\n", " * new_school (new_school) <U13 'Essex College' 'Moordale'\n", "Data variables:\n", " obs (chain, draw, new_school) float64 2.041 -2.556 ... -0.2822\n", "Attributes: (2)
<xarray.Dataset>\n", "Dimensions: (chain: 4, draw: 500, school: 8)\n", "Coordinates:\n", " * chain (chain) int64 0 1 2 3\n", " * draw (draw) int64 0 1 2 3 4 5 6 7 8 ... 492 493 494 495 496 497 498 499\n", " * school (school) <U16 'Choate' 'Deerfield' ... "St. Paul's" 'Mt. Hermon'\n", "Data variables:\n", " obs (chain, draw, school) float64 ...\n", "Attributes: (4)
<xarray.Dataset>\n", "Dimensions: (chain: 4, draw: 500)\n", "Coordinates:\n", " * chain (chain) int64 0 1 2 3\n", " * draw (draw) int64 0 1 2 3 4 5 6 ... 494 495 496 497 498 499\n", "Data variables: (12/16)\n", " max_energy_error (chain, draw) float64 ...\n", " energy_error (chain, draw) float64 ...\n", " lp (chain, draw) float64 ...\n", " index_in_trajectory (chain, draw) int64 ...\n", " acceptance_rate (chain, draw) float64 ...\n", " diverging (chain, draw) bool ...\n", " ... ...\n", " smallest_eigval (chain, draw) float64 ...\n", " step_size_bar (chain, draw) float64 ...\n", " step_size (chain, draw) float64 ...\n", " energy (chain, draw) float64 ...\n", " tree_depth (chain, draw) int64 ...\n", " perf_counter_diff (chain, draw) float64 ...\n", "Attributes: (6)
<xarray.Dataset>\n", "Dimensions: (draw: 500, school: 8)\n", "Coordinates:\n", " * draw (draw) int64 0 1 2 3 4 5 6 7 8 ... 492 493 494 495 496 497 498 499\n", " * school (school) <U16 'Choate' 'Deerfield' ... "St. Paul's" 'Mt. Hermon'\n", "Data variables:\n", " tau (draw) float64 1.941 3.388 4.208 5.687 ... 0.8353 0.06893 2.145\n", " theta (draw, school) float64 4.866 4.59 -0.7404 ... 3.33 -2.031 6.045\n", " mu (draw) float64 3.903 3.915 -1.751 2.595 ... -2.294 0.7908 2.869
<xarray.Dataset>\n", "Dimensions: (chain: 1, draw: 500, school: 8)\n", "Coordinates:\n", " * chain (chain) int64 0\n", " * draw (draw) int64 0 1 2 3 4 5 6 7 8 ... 492 493 494 495 496 497 498 499\n", " * school (school) <U16 'Choate' 'Deerfield' ... "St. Paul's" 'Mt. Hermon'\n", "Data variables:\n", " obs (chain, draw, school) float64 ...\n", "Attributes: (4)
<xarray.Dataset>\n", "Dimensions: (school: 8)\n", "Coordinates:\n", " * school (school) <U16 'Choate' 'Deerfield' ... "St. Paul's" 'Mt. Hermon'\n", "Data variables:\n", " obs (school) float64 ...\n", "Attributes: (4)
<xarray.Dataset>\n", "Dimensions: (school: 8)\n", "Coordinates:\n", " * school (school) <U16 'Choate' 'Deerfield' ... "St. Paul's" 'Mt. Hermon'\n", "Data variables:\n", " scores (school) float64 ...\n", "Attributes: (4)
<xarray.Dataset>\n", "Dimensions: (draw: 500, school: 8, school_bis: 8)\n", "Coordinates:\n", " * draw (draw) int64 0 1 2 3 4 5 6 ... 494 495 496 497 498 499\n", " * school (school) <U16 'Choate' 'Deerfield' ... 'Mt. Hermon'\n", " * school_bis (school_bis) <U16 'Choate' 'Deerfield' ... 'Mt. Hermon'\n", "Data variables:\n", " mu (draw) float64 5.974 5.096 7.177 ... 3.284 4.739 3.146\n", " theta (draw, school) float64 9.519 5.554 6.118 ... 5.595 3.773\n", " tau (draw) float64 4.068 3.156 3.603 ... 2.725 3.225 2.979\n", " log_tau (draw) float64 1.322 1.118 1.234 ... 0.958 1.035 0.9508\n", " mlogtau (draw) float64 nan nan nan nan ... 0.993 1.002 1.01 1.021\n", " theta_school_diff (draw, school, school_bis) float64 0.0 3.965 ... 0.0
<xarray.Dataset>\n", "Dimensions: (chain: 4, draw: 500, school: 8, Upper: 8)\n", "Coordinates:\n", " * chain (chain) int64 0 1 2 3\n", " * draw (draw) int64 0 1 2 3 4 5 6 7 8 ... 492 493 494 495 496 497 498 499\n", " * school (school) <U16 'Choate' 'Deerfield' ... "St. Paul's" 'Mt. Hermon'\n", " upper (Upper) <U16 'CHOATE' 'DEERFIELD' ... "ST. PAUL'S" 'MT. HERMON'\n", "Dimensions without coordinates: Upper\n", "Data variables:\n", " obs (chain, draw, school) float64 ...\n", "Attributes: (4)
<xarray.Dataset>\n", "Dimensions: (chain: 4, draw: 500, new_school: 2)\n", "Coordinates:\n", " * chain (chain) int64 0 1 2 3\n", " * draw (draw) int64 0 1 2 3 4 5 6 7 ... 492 493 494 495 496 497 498 499\n", " * new_school (new_school) <U13 'Essex College' 'Moordale'\n", "Data variables:\n", " obs (chain, draw, new_school) float64 2.041 -2.556 ... -0.2822\n", "Attributes: (2)
<xarray.Dataset>\n", "Dimensions: (chain: 4, draw: 500, school: 8)\n", "Coordinates:\n", " * chain (chain) int64 0 1 2 3\n", " * draw (draw) int64 0 1 2 3 4 5 6 7 8 ... 492 493 494 495 496 497 498 499\n", " * school (school) <U16 'Choate' 'Deerfield' ... "St. Paul's" 'Mt. Hermon'\n", "Data variables:\n", " obs (chain, draw, school) float64 ...\n", "Attributes: (4)
<xarray.Dataset>\n", "Dimensions: (chain: 4, draw: 500)\n", "Coordinates:\n", " * chain (chain) int64 0 1 2 3\n", " * draw (draw) int64 0 1 2 3 4 5 6 ... 494 495 496 497 498 499\n", "Data variables: (12/16)\n", " max_energy_error (chain, draw) float64 ...\n", " energy_error (chain, draw) float64 ...\n", " lp (chain, draw) float64 ...\n", " index_in_trajectory (chain, draw) int64 ...\n", " acceptance_rate (chain, draw) float64 ...\n", " diverging (chain, draw) bool ...\n", " ... ...\n", " smallest_eigval (chain, draw) float64 ...\n", " step_size_bar (chain, draw) float64 ...\n", " step_size (chain, draw) float64 ...\n", " energy (chain, draw) float64 ...\n", " tree_depth (chain, draw) int64 ...\n", " perf_counter_diff (chain, draw) float64 ...\n", "Attributes: (6)
<xarray.Dataset>\n", "Dimensions: (draw: 500, school: 8)\n", "Coordinates:\n", " * draw (draw) int64 0 1 2 3 4 5 6 7 8 ... 492 493 494 495 496 497 498 499\n", " * school (school) <U16 'Choate' 'Deerfield' ... "St. Paul's" 'Mt. Hermon'\n", "Data variables:\n", " tau (draw) float64 1.941 3.388 4.208 5.687 ... 0.8353 0.06893 2.145\n", " theta (draw, school) float64 4.866 4.59 -0.7404 ... 3.33 -2.031 6.045\n", " mu (draw) float64 3.903 3.915 -1.751 2.595 ... -2.294 0.7908 2.869
<xarray.Dataset>\n", "Dimensions: (chain: 1, draw: 500, school: 8, Upper: 8)\n", "Coordinates:\n", " * chain (chain) int64 0\n", " * draw (draw) int64 0 1 2 3 4 5 6 7 8 ... 492 493 494 495 496 497 498 499\n", " * school (school) <U16 'Choate' 'Deerfield' ... "St. Paul's" 'Mt. Hermon'\n", " upper (Upper) <U16 'CHOATE' 'DEERFIELD' ... "ST. PAUL'S" 'MT. HERMON'\n", "Dimensions without coordinates: Upper\n", "Data variables:\n", " obs (chain, draw, school) float64 ...\n", "Attributes: (4)
<xarray.Dataset>\n", "Dimensions: (school: 8, Upper: 8)\n", "Coordinates:\n", " * school (school) <U16 'Choate' 'Deerfield' ... "St. Paul's" 'Mt. Hermon'\n", " upper (Upper) <U16 'CHOATE' 'DEERFIELD' ... "ST. PAUL'S" 'MT. HERMON'\n", "Dimensions without coordinates: Upper\n", "Data variables:\n", " obs (school) float64 28.0 8.0 -3.0 7.0 -1.0 1.0 18.0 12.0\n", "Attributes: (4)
<xarray.Dataset>\n", "Dimensions: (school: 8)\n", "Coordinates:\n", " * school (school) <U16 'Choate' 'Deerfield' ... "St. Paul's" 'Mt. Hermon'\n", "Data variables:\n", " scores (school) float64 ...\n", "Attributes: (4)