{ "cells": [ { "cell_type": "markdown", "metadata": {}, "source": [ "Original question: https://discourse.pymc.io/t/11-16-rethinking-code/5181" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "# Introduction\n", "Contains the observed data and the sampling of the PyMC3 model" ] }, { "cell_type": "code", "execution_count": 1, "metadata": {}, "outputs": [], "source": [ "import arviz as az\n", "import matplotlib.pyplot as plt\n", "import numpy as np\n", "import pandas as pd\n", "import pymc3 as pm\n", "import theano.tensor as tt\n", "import warnings\n", "\n", "from scipy import stats\n", "from scipy.special import expit as logistic\n", "from scipy.special import softmax\n", "\n", "warnings.simplefilter(action=\"ignore\", category=FutureWarning)\n", "RANDOM_SEED = 8927\n", "np.random.seed(286)" ] }, { "cell_type": "code", "execution_count": 2, "metadata": {}, "outputs": [], "source": [ "az.style.use(\"arviz-darkgrid\")\n", "az.rcParams[\"stats.hdi_prob\"] = 0.89 # name was changed in latest ArviZ" ] }, { "cell_type": "code", "execution_count": 3, "metadata": {}, "outputs": [ { "data": { "text/html": [ "
\n", "\n", "\n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", "
actorprosoc_leftconditiontreatment
00000
10000
20101
30000
40101
...............
4996113
5006113
5016012
5026012
5036012
\n", "

504 rows × 4 columns

\n", "
" ], "text/plain": [ " actor prosoc_left condition treatment\n", "0 0 0 0 0\n", "1 0 0 0 0\n", "2 0 1 0 1\n", "3 0 0 0 0\n", "4 0 1 0 1\n", ".. ... ... ... ...\n", "499 6 1 1 3\n", "500 6 1 1 3\n", "501 6 0 1 2\n", "502 6 0 1 2\n", "503 6 0 1 2\n", "\n", "[504 rows x 4 columns]" ] }, "execution_count": 3, "metadata": {}, "output_type": "execute_result" } ], "source": [ "d = pd.read_csv(\n", " \"https://raw.githubusercontent.com/pymc-devs/resources/master/Rethinking_2/Data/chimpanzees.csv\", \n", " sep=\";\"\n", ")\n", "# we change \"actor\" to zero-index\n", "d.actor = d.actor - 1\n", "d[\"treatment\"] = d.prosoc_left + 2 * d.condition\n", "d[[\"actor\", \"prosoc_left\", \"condition\", \"treatment\"]]" ] }, { "cell_type": "code", "execution_count": 4, "metadata": {}, "outputs": [], "source": [ "actor_idx, actors = pd.factorize(d.actor)\n", "treat_idx, treatments = pd.factorize(d.treatment)" ] }, { "cell_type": "code", "execution_count": 5, "metadata": {}, "outputs": [ { "name": "stderr", "output_type": "stream", "text": [ "Auto-assigning NUTS sampler...\n", "Initializing NUTS using jitter+adapt_diag...\n", "Multiprocess sampling (4 chains in 4 jobs)\n", "NUTS: [b, a]\n", "Sampling 4 chains, 0 divergences: 100%|██████████| 4000/4000 [00:01<00:00, 2388.27draws/s]\n" ] } ], "source": [ "with pm.Model() as m11_4:\n", " a = pm.Normal(\"a\", 0.0, 1.5, shape=len(actors))\n", " b = pm.Normal(\"b\", 0.0, 0.5, shape=len(treatments))\n", "\n", " actor_id = pm.intX(pm.Data(\"actor_id\", actor_idx))\n", " treat_id = pm.intX(pm.Data(\"treat_id\", treat_idx))\n", " p = pm.Deterministic(\"p\", pm.math.invlogit(a[actor_id] + b[treat_id]))\n", "\n", " pulled_left = pm.Binomial(\"pulled_left\", 1, p, observed=d.pulled_left)\n", "\n", " trace_11_4 = pm.sample(random_seed=RANDOM_SEED)" ] }, { "cell_type": "code", "execution_count": 6, "metadata": {}, "outputs": [ { "data": { "text/html": [ "
\n", "\n", "\n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", "
meansdhdi_5.5%hdi_94.5%mcse_meanmcse_sdess_meaness_sdess_bulkess_tailr_hat
a[0]-0.440.32-0.970.050.010.01696.64696.64697.771204.461.01
a[1]3.900.762.825.270.020.011685.561590.311734.731307.421.00
a[2]-0.750.33-1.31-0.250.010.01684.50684.50684.25951.561.01
a[3]-0.740.33-1.28-0.240.010.01746.43746.43758.531041.131.01
a[4]-0.440.33-0.920.110.010.01672.00672.00673.091275.211.00
a[5]0.490.34-0.031.030.010.01623.04623.04622.091042.271.01
a[6]1.950.411.342.650.010.01978.12974.13995.211331.111.00
b[treatment 1]-0.040.29-0.530.380.010.01591.45591.45590.001137.321.01
b[treatment 2]0.480.290.030.950.010.01637.00637.00637.811418.871.01
b[treatment 3]-0.390.28-0.840.050.010.01596.72596.72595.32989.791.01
b[treatment 4]0.360.29-0.070.820.010.01580.24580.24577.46931.291.01
\n", "
" ], "text/plain": [ " mean sd hdi_5.5% hdi_94.5% mcse_mean mcse_sd ess_mean \\\n", "a[0] -0.44 0.32 -0.97 0.05 0.01 0.01 696.64 \n", "a[1] 3.90 0.76 2.82 5.27 0.02 0.01 1685.56 \n", "a[2] -0.75 0.33 -1.31 -0.25 0.01 0.01 684.50 \n", "a[3] -0.74 0.33 -1.28 -0.24 0.01 0.01 746.43 \n", "a[4] -0.44 0.33 -0.92 0.11 0.01 0.01 672.00 \n", "a[5] 0.49 0.34 -0.03 1.03 0.01 0.01 623.04 \n", "a[6] 1.95 0.41 1.34 2.65 0.01 0.01 978.12 \n", "b[treatment 1] -0.04 0.29 -0.53 0.38 0.01 0.01 591.45 \n", "b[treatment 2] 0.48 0.29 0.03 0.95 0.01 0.01 637.00 \n", "b[treatment 3] -0.39 0.28 -0.84 0.05 0.01 0.01 596.72 \n", "b[treatment 4] 0.36 0.29 -0.07 0.82 0.01 0.01 580.24 \n", "\n", " ess_sd ess_bulk ess_tail r_hat \n", "a[0] 696.64 697.77 1204.46 1.01 \n", "a[1] 1590.31 1734.73 1307.42 1.00 \n", "a[2] 684.50 684.25 951.56 1.01 \n", "a[3] 746.43 758.53 1041.13 1.01 \n", "a[4] 672.00 673.09 1275.21 1.00 \n", "a[5] 623.04 622.09 1042.27 1.01 \n", "a[6] 974.13 995.21 1331.11 1.00 \n", "b[treatment 1] 591.45 590.00 1137.32 1.01 \n", "b[treatment 2] 637.00 637.81 1418.87 1.01 \n", "b[treatment 3] 596.72 595.32 989.79 1.01 \n", "b[treatment 4] 580.24 577.46 931.29 1.01 " ] }, "execution_count": 6, "metadata": {}, "output_type": "execute_result" } ], "source": [ "treatment_names = [f\"treatment {i+1}\" for i in treatments]\n", "# I have given some names to the treatments to show ArviZ capabilities with \n", "# labels thanks to xarray\n", "dims = {\n", " \"a\": [\"actor\"], \n", " \"b\": [\"treatment\"],\n", " \"actor_id\": [\"experiment\"],\n", " \"treat_id\": [\"experiment\"],\n", " \"p\": [\"experiment\"],\n", " \"pulled_left\": [\"experiment\"]\n", "}\n", "coords = {\"actor\": actors, \"treatment\": treatment_names} \n", "\n", "# also, it is highly recommended to either pass the model instance to from_pymc3\n", "# or to call it from the corresponding model context\n", "idata_11_4 = az.from_pymc3(trace_11_4, model=m11_4, dims=dims, coords=coords)\n", "az.summary(idata_11_4, var_names=[\"a\", \"b\"], round_to=2)" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "_Note: I am using this notebook also as a test for [arviz-devs/arviz#1201](https://github.com/arviz-devs/arviz/pull/1201) so I have coordinate values instead of indexes shown in the summary. This has no relation to the current quesion, it is only to prevent surprises when running the notebook on ArviZ<=0.8.3_" ] }, { "cell_type": "code", "execution_count": 7, "metadata": {}, "outputs": [ { "data": { "text/plain": [ "(4, 500, 504)" ] }, "execution_count": 7, "metadata": {}, "output_type": "execute_result" } ], "source": [ "idata_11_4.posterior[\"p\"].shape" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "We have 504 different $p$ but $n_{actors} \\times n_{treatments} = 28$?!?! How do we get the 28 $p$ corresponding the each actor-treatment combination?" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "# To the question at hand\n", "As we have just seen, our values for $p$ correspond to each experiment, and there are 504 of them. \n", "\n", "Each experiment was done for a specific combination of actor and treatment and different experiments correspond to the same combination of actor and treatment. Thus, to get the $p$ corresponding to each of the 28 actor-treatment combinations we have (at least) 3 options (we'll focus on the 2nd and 3dr ones):\n", "\n", "1. Group-by and average the original 504 $p$. This can be straighforward if there are the same number of experiments for each actor-treatment (a reshape could be enough if properly ordered) but with unsorted and ragged data things can get out of control.\n", "1. pm.sample_posterior_predictive: [pm.Data](https://docs.pymc.io/api/model.html#pymc3.model.set_data) containers are thought to allow changing the data in the model and get predictions for new values for example. In this particular case, we are modifying the data to contain only one experiment per actor-treatment combination which automatically gets us to the case in option 1 where only a reshape is needed.\n", "3. Calculate $p$ manually from $a$ and $b$. We can manually calculate $logistic(a+b)$ for every combination of actor-treatment\n", "\n", "It should be noted that even though we are using pm.sample_posterior_predictive in option 2, we are not getting samples from the posterior predictive. We are using it to cleverly reevaluate a deterministic function from the posterior. \n", "\n", "We would be sampling from the posterior predictive if we obtained values from pulled_left. The obtained values could then be averaged to _estimate_ the probability, but given that we do know the probability it does not make much sense to _estimate_ it.\n" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "## pm.sample_posterior_predictive version" ] }, { "cell_type": "code", "execution_count": 8, "metadata": {}, "outputs": [ { "name": "stderr", "output_type": "stream", "text": [ "100%|██████████| 2000/2000 [00:00<00:00, 3839.00it/s]\n" ] } ], "source": [ "with m11_4:\n", " pm.set_data({\"actor_id\": np.repeat(range(7), 4), \"treat_id\": list(range(4)) * 7})\n", " p_post = pm.sample_posterior_predictive(\n", " trace_11_4, random_seed=RANDOM_SEED, var_names=[\"p\"]\n", " )[\"p\"]\n", "p_mu = p_post.mean(0).reshape((7, 4))" ] }, { "cell_type": "code", "execution_count": 9, "metadata": {}, "outputs": [ { "data": { "text/plain": [ "array([[0.38273298, 0.50794882, 0.3064966 , 0.48028027],\n", " [0.97383758, 0.9842027 , 0.96366044, 0.98230814],\n", " [0.31552197, 0.43399235, 0.24723306, 0.40704301],\n", " [0.31658471, 0.43514105, 0.24811081, 0.40830217],\n", " [0.38371248, 0.50898154, 0.30735674, 0.4813107 ],\n", " [0.60644722, 0.71919498, 0.52349316, 0.69665124],\n", " [0.8650691 , 0.91412716, 0.82050517, 0.90508358]])" ] }, "execution_count": 9, "metadata": {}, "output_type": "execute_result" } ], "source": [ "p_mu" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "## Manual version\n", "The manual version will use ArviZ and xarray to showcase its capabilities. The posterior group in the InferenceData object is an xarray.Dataset. As it can be seen below, the dataset stores both the data and its dimensions and coordinate values:" ] }, { "cell_type": "code", "execution_count": 10, "metadata": {}, "outputs": [ { "data": { "text/html": [ "
\n", "
xarray.Dataset
• actor: 7
• chain: 4
• draw: 500
• experiment: 504
• treatment: 4
• chain
(chain)
int64
0 1 2 3
array([0, 1, 2, 3])
• draw
(draw)
int64
0 1 2 3 4 5 ... 495 496 497 498 499
array([  0,   1,   2, ..., 497, 498, 499])
• actor
(actor)
int64
0 1 2 3 4 5 6
array([0, 1, 2, 3, 4, 5, 6])
• treatment
(treatment)
<U11
'treatment 1' ... 'treatment 4'
array(['treatment 1', 'treatment 2', 'treatment 3', 'treatment 4'],\n",
"      dtype='<U11')
• experiment
(experiment)
int64
0 1 2 3 4 5 ... 499 500 501 502 503
array([  0,   1,   2, ..., 501, 502, 503])
• a
(chain, draw, actor)
float64
0.135 3.439 ... 0.8038 2.469
array([[[ 0.1350494 ,  3.43949205, -0.12937185, ...,  0.03736965,\n",
"          1.13246861,  2.41149076],\n",
"        [-0.53867798,  5.54241401, -0.86097509, ..., -0.49340846,\n",
"          0.51538093,  2.39456749],\n",
"        [-0.6464498 ,  2.63123411, -0.15443548, ..., -0.12293604,\n",
"          0.72261353,  1.76719445],\n",
"        ...,\n",
"        [-0.22109826,  3.77049511, -0.65876488, ..., -0.36966607,\n",
"          0.3325513 ,  1.70170163],\n",
"        [-0.46268774,  4.32909982, -0.51364639, ..., -0.28914722,\n",
"          0.34148056,  2.02406726],\n",
"        [-0.79424648,  3.23830151, -1.2127418 , ..., -0.56576385,\n",
"          0.14399398,  2.40274617]],\n",
"\n",
"       [[-0.44553248,  3.42886858, -0.63826118, ..., -0.67836183,\n",
"          0.81808415,  1.80385049],\n",
"        [-0.28375707,  3.41254227, -0.52666946, ..., -0.65795236,\n",
"          0.72161261,  2.03624599],\n",
"        [-0.24907328,  3.5322941 , -0.78412297, ..., -0.63089803,\n",
"          1.33693092,  1.96060643],\n",
"        ...,\n",
"        [-0.15288734,  4.08486494,  0.26401406, ..., -0.49760689,\n",
"          1.507072  ,  1.88346066],\n",
"        [-0.42494098,  4.82477725, -0.79238959, ...,  0.09061211,\n",
"          0.31683397,  2.04605659],\n",
"        [ 0.46118619,  4.64381499,  0.07416212, ...,  0.33530701,\n",
"          1.21827518,  2.91216765]],\n",
"\n",
"       [[-0.78547128,  4.7450789 , -0.78546263, ..., -0.80163116,\n",
"          0.15630192,  1.24127972],\n",
"        [-0.33331993,  2.90132854, -0.76646069, ..., -0.30024591,\n",
"          0.74294793,  2.61416152],\n",
"        [-0.39723684,  3.41333113, -0.39619374, ..., -0.96907126,\n",
"          0.10858268,  1.1236944 ],\n",
"        ...,\n",
"        [ 0.05353087,  4.87493456,  0.11899031, ...,  0.15286503,\n",
"          0.89845667,  1.89079282],\n",
"        [-0.17776394,  5.24086845, -0.13968179, ..., -0.40650101,\n",
"          0.85596658,  2.32740041],\n",
"        [-0.06807385,  3.49187068, -0.7569612 , ...,  0.12827456,\n",
"          0.68496056,  2.1930579 ]],\n",
"\n",
"       [[-0.72787785,  3.3199782 , -0.80540404, ..., -0.57141605,\n",
"          0.17284305,  1.1029963 ],\n",
"        [-0.44173558,  4.22527233, -1.24332166, ..., -0.65391359,\n",
"         -0.36336629,  2.39538558],\n",
"        [-0.3523625 ,  3.91915261, -1.2200635 , ..., -0.85952723,\n",
"         -0.54765908,  2.52172622],\n",
"        ...,\n",
"        [-0.5660457 ,  3.07675958, -0.39877395, ..., -0.48418973,\n",
"          0.88115401,  2.39811526],\n",
"        [-0.57515718,  4.85570589, -0.47318295, ..., -0.02808618,\n",
"          0.61239922,  1.81332135],\n",
"        [-0.32259427,  2.92729824, -0.54600485, ..., -0.38426948,\n",
"          0.80376838,  2.46907673]]])
• b
(chain, draw, treatment)
float64
-0.3593 0.08189 ... -0.4503 0.3017
array([[[-0.35929964,  0.08188989, -0.90768709, -0.31891759],\n",
"        [-0.18308382,  0.38016992, -0.46203453,  0.55845587],\n",
"        [-0.40881454,  0.12135358, -0.43520694,  0.18602932],\n",
"        ...,\n",
"        [-0.13702956,  0.25173945, -0.42576259,  0.69042055],\n",
"        [-0.16249926,  0.23940038, -0.59371464,  0.60846425],\n",
"        [-0.15320059,  1.0583223 , -0.53275706,  0.76881312]],\n",
"\n",
"       [[-0.4386647 ,  0.43428959, -0.69013596,  0.5368523 ],\n",
"        [-0.26414823,  0.30762992, -0.53558744,  0.52895948],\n",
"        [-0.22472049,  0.7230068 , -0.79167565,  0.47707052],\n",
"        ...,\n",
"        [-0.20719957,  0.03885995, -1.01523696, -0.38181753],\n",
"        [-0.52578112,  0.3888622 , -0.33594527,  0.36942624],\n",
"        [-1.17373519, -0.19527191, -0.96670953, -0.23657849]],\n",
"\n",
"       [[ 0.21501781,  0.64151466, -0.2640097 ,  0.57598384],\n",
"        [ 0.0341824 ,  0.16604603, -0.65063801,  0.26403522],\n",
"        [-0.01247477,  0.82985674, -0.19416263,  0.54055417],\n",
"        ...,\n",
"        [-0.397266  ,  0.2877604 , -0.58344193,  0.12683765],\n",
"        [-0.5483128 ,  0.41723657, -0.50482686, -0.04789725],\n",
"        [-0.17635197, -0.10309123, -0.8953712 ,  0.13159195]],\n",
"\n",
"       [[ 0.05935307,  1.01295935,  0.07383078,  0.56308356],\n",
"        [ 0.29942449,  0.78510956, -0.42217475,  0.66064565],\n",
"        [ 0.07936967,  0.63896814, -0.58651151,  0.71442086],\n",
"        ...,\n",
"        [-0.00976328,  0.44357886, -0.36715989,  0.28928022],\n",
"        [-0.52414581,  0.20999093, -0.66477251,  0.08737833],\n",
"        [-0.1322785 ,  0.55515128, -0.45029006,  0.30174006]]])
• p
(chain, draw, experiment)
float64
0.4442 0.4442 ... 0.8828 0.8828
array([[[0.4441712 , 0.4441712 , 0.55402312, ..., 0.81814109,\n",
"         0.81814109, 0.81814109],\n",
"        [0.32700514, 0.32700514, 0.46045575, ..., 0.87352952,\n",
"         0.87352952, 0.87352952],\n",
"        [0.25821549, 0.25821549, 0.37166134, ..., 0.7911692 ,\n",
"         0.7911692 , 0.7911692 ],\n",
"        ...,\n",
"        [0.41141284, 0.41141284, 0.5076597 , ..., 0.78175772,\n",
"         0.78175772, 0.78175772],\n",
"        [0.34860267, 0.34860267, 0.44440894, ..., 0.80695625,\n",
"         0.80695625, 0.80695625],\n",
"        [0.27939853, 0.27939853, 0.56563795, ..., 0.86645702,\n",
"         0.86645702, 0.86645702]],\n",
"\n",
"       [[0.29230878, 0.29230878, 0.49718931, ..., 0.75282097,\n",
"         0.75282097, 0.75282097],\n",
"        [0.36635053, 0.36635053, 0.50596793, ..., 0.81767268,\n",
"         0.81767268, 0.81767268],\n",
"        [0.3837187 , 0.3837187 , 0.61631435, ..., 0.7629517 ,\n",
"         0.7629517 , 0.7629517 ],\n",
"        ...,\n",
"        [0.41093853, 0.41093853, 0.471524  , ..., 0.70437595,\n",
"         0.70437595, 0.70437595],\n",
"        [0.27873963, 0.27873963, 0.49098128, ..., 0.84685072,\n",
"         0.84685072, 0.84685072],\n",
"        [0.32903585, 0.32903585, 0.56608959, ..., 0.87495055,\n",
"         0.87495055, 0.87495055]],\n",
"\n",
"       [[0.3611322 , 0.3611322 , 0.46407287, ..., 0.72656619,\n",
"         0.72656619, 0.72656619],\n",
"        [0.42576834, 0.42576834, 0.45827876, ..., 0.87691377,\n",
"         0.87691377, 0.87691377],\n",
"        [0.39898127, 0.39898127, 0.6064991 , ..., 0.71698028,\n",
"         0.71698028, 0.71698028],\n",
"        ...,\n",
"        [0.41490246, 0.41490246, 0.58450415, ..., 0.78706953,\n",
"         0.78706953, 0.78706953],\n",
"        [0.32605625, 0.32605625, 0.55958368, ..., 0.86087465,\n",
"         0.86087465, 0.86087465],\n",
"        [0.43919597, 0.43919597, 0.4573129 , ..., 0.7854454 ,\n",
"         0.7854454 , 0.7854454 ]],\n",
"\n",
"       [[0.33882725, 0.33882725, 0.57079158, ..., 0.76437682,\n",
"         0.76437682, 0.76437682],\n",
"        [0.46448215, 0.46448215, 0.58500987, ..., 0.87795557,\n",
"         0.87795557, 0.87795557],\n",
"        [0.43217251, 0.43217251, 0.57116494, ..., 0.87382549,\n",
"         0.87382549, 0.87382549],\n",
"        ...,\n",
"        [0.35989752, 0.35989752, 0.4694215 , ..., 0.88400907,\n",
"         0.88400907, 0.88400907],\n",
"        [0.24987052, 0.24987052, 0.40970954, ..., 0.75924576,\n",
"         0.75924576, 0.75924576],\n",
"        [0.38820285, 0.38820285, 0.55787864, ..., 0.88275549,\n",
"         0.88275549, 0.88275549]]])
• created_at :
2020-06-04T22:45:38.724507
arviz_version :
0.8.3
inference_library :
pymc3
inference_library_version :
3.8
" ], "text/plain": [ "\n", "Dimensions: (actor: 7, chain: 4, draw: 500, experiment: 504, treatment: 4)\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", " * actor (actor) int64 0 1 2 3 4 5 6\n", " * treatment (treatment) \n", "
xarray.DataArray
• chain: 4
• draw: 500
• actor: 7
• treatment: 4
• 0.4442 0.554 0.3159 0.4542 0.9561 ... 0.9119 0.9537 0.8828 0.9411
array([[[[0.4441712 , 0.55402312, 0.3159088 , 0.45416202],\n",
"         [0.95606827, 0.97129007, 0.92634161, 0.95773349],\n",
"         [0.38020658, 0.48813174, 0.26171787, 0.38976755],\n",
"         ...,\n",
"         [0.42020547, 0.5297796 , 0.29518825, 0.43007432],\n",
"         [0.68420601, 0.77106923, 0.55595995, 0.69286569],\n",
"         [0.88616883, 0.92367647, 0.81814109, 0.89017923]],\n",
"\n",
"        [[0.32700514, 0.46045575, 0.26880136, 0.50494431],\n",
"         [0.99531797, 0.99732888, 0.99382087, 0.99776409],\n",
"         [0.26036758, 0.38206201, 0.21031801, 0.42494175],\n",
"         ...,\n",
"         [0.33704464, 0.47172058, 0.27779151, 0.51625612],\n",
"         [0.58231819, 0.71003434, 0.51333344, 0.74532588],\n",
"         [0.90127602, 0.94129532, 0.87352952, 0.95040619]],\n",
"\n",
"        [[0.25821549, 0.37166134, 0.25319262, 0.38688608],\n",
"         [0.90224481, 0.94005933, 0.89989218, 0.94360161],\n",
"         [0.3627958 , 0.49173028, 0.3567169 , 0.5078978 ],\n",
"         ...,\n",
"         [0.37010869, 0.49960439, 0.36397725, 0.51576809],\n",
"         [0.57781228, 0.69930008, 0.5713611 , 0.71272237],\n",
"         [0.79549626, 0.86858989, 0.7911692 , 0.87579774]],\n",
"\n",
"        ...,\n",
"\n",
"        [[0.41141284, 0.5076597 , 0.34369729, 0.61522334],\n",
"         [0.97425583, 0.98240233, 0.96593192, 0.98858014],\n",
"         [0.31092585, 0.39962558, 0.25265019, 0.50791326],\n",
"         ...,\n",
"         [0.37596846, 0.47055246, 0.31100422, 0.57950811],\n",
"         [0.54872531, 0.64205411, 0.47671404, 0.73555108],\n",
"         [0.82702274, 0.87582137, 0.78175772, 0.9162246 ]],\n",
"\n",
"        [[0.34860267, 0.44440894, 0.25799757, 0.53637973],\n",
"         [0.98473185, 0.989733  , 0.97669224, 0.99287902],\n",
"         [0.3371221 , 0.431865  , 0.2483632 , 0.52368672],\n",
"         ...,\n",
"         [0.38896937, 0.48756585, 0.29258508, 0.5791578 ],\n",
"         [0.54462626, 0.64127009, 0.43727369, 0.72110408],\n",
"         [0.86547961, 0.90580591, 0.80695625, 0.93292613]],\n",
"\n",
"        [[0.27939853, 0.56563795, 0.20965545, 0.493642  ],\n",
"         [0.95627397, 0.98656842, 0.93735302, 0.98213902],\n",
"         [0.20327621, 0.46147166, 0.14861582, 0.39080524],\n",
"         ...,\n",
"         [0.32762106, 0.62070895, 0.25001713, 0.55058863],\n",
"         [0.49769836, 0.76893658, 0.4040151 , 0.71357424],\n",
"         [0.90461133, 0.96955952, 0.86645702, 0.95974986]]],\n",
"\n",
"\n",
"       [[[0.29230878, 0.49718931, 0.24311653, 0.5228141 ],\n",
"         [0.9521296 , 0.97943043, 0.93927385, 0.98139822],\n",
"         [0.25408821, 0.44918316, 0.20942462, 0.47466948],\n",
"         ...,\n",
"         [0.24656325, 0.43928306, 0.20286266, 0.46468154],\n",
"         [0.59373307, 0.7777105 , 0.53194348, 0.79493551],\n",
"         [0.79660123, 0.9036226 , 0.75282097, 0.91219239]],\n",
"\n",
"        [[0.36635053, 0.50596793, 0.30590282, 0.5609953 ],\n",
"         [0.9588454 , 0.9763434 , 0.9466954 , 0.98095088],\n",
"         [0.31199312, 0.44545801, 0.25687839, 0.5005725 ],\n",
"         ...,\n",
"         [0.28453008, 0.41330423, 0.23262644, 0.46779642],\n",
"         [0.61241248, 0.73676902, 0.54637264, 0.77739888],\n",
"         [0.85471835, 0.91244622, 0.81767268, 0.92858841]],\n",
"\n",
"        [[0.3837187 , 0.61631435, 0.26100551, 0.55675367],\n",
"         [0.96468772, 0.98600969, 0.93938132, 0.98217845],\n",
"         [0.26720625, 0.48472571, 0.17139132, 0.42383437],\n",
"         ...,\n",
"         [0.29825557, 0.52301093, 0.19425843, 0.46161878],\n",
"         [0.75254097, 0.88694793, 0.63303408, 0.85984479],\n",
"         [0.85016375, 0.93605275, 0.7629517 , 0.91965561]],\n",
"\n",
"        ...,\n",
"\n",
"        [[0.41093853, 0.471524  , 0.23719419, 0.36942022],\n",
"         [0.97972067, 0.98407364, 0.9556224 , 0.97594463],\n",
"         [0.5141998 , 0.57514494, 0.3205549 , 0.47058315],\n",
"         ...,\n",
"         [0.33074744, 0.38728313, 0.18051772, 0.29329707],\n",
"         [0.78581351, 0.8243254 , 0.62053862, 0.75496207],\n",
"         [0.8424088 , 0.87239699, 0.70437595, 0.81781942]],\n",
"\n",
"        [[0.27873963, 0.49098128, 0.31845388, 0.48612488],\n",
"         [0.98659982, 0.99458762, 0.98889104, 0.99448198],\n",
"         [0.2111228 , 0.40046514, 0.24446853, 0.39580787],\n",
"         ...,\n",
"         [0.3928927 , 0.61762373, 0.4389725 , 0.61302327],\n",
"         [0.44795244, 0.66944948, 0.49522232, 0.66513448],\n",
"         [0.82057904, 0.91945157, 0.84685072, 0.91800035]],\n",
"\n",
"        [[0.32903585, 0.56608959, 0.37624354, 0.55591704],\n",
"         [0.96982435, 0.98843961, 0.97532802, 0.98795796],\n",
"         [0.2498199 , 0.46975951, 0.29058441, 0.45948493],\n",
"         ...,\n",
"         [0.30186593, 0.53495168, 0.34719259, 0.5246621 ],\n",
"         [0.51113316, 0.73555719, 0.56256182, 0.72744475],\n",
"         [0.85048785, 0.93801629, 0.87495055, 0.93557076]]],\n",
"\n",
"\n",
"       [[[0.3611322 , 0.46407287, 0.25932478, 0.44781883],\n",
"         [0.99303658, 0.99544332, 0.98880544, 0.99513622],\n",
"         [0.36113419, 0.46407502, 0.25932644, 0.44782097],\n",
"         ...,\n",
"         [0.35741229, 0.46005618, 0.25623296, 0.44382632],\n",
"         [0.59177783, 0.68950723, 0.47309906, 0.67530667],\n",
"         [0.81096574, 0.86793177, 0.72656619, 0.86023745]],\n",
"\n",
"        [[0.42576834, 0.45827876, 0.27210715, 0.48268575],\n",
"         [0.94957421, 0.95552674, 0.90471008, 0.95950985],\n",
"         [0.32469497, 0.35424883, 0.19511682, 0.37697084],\n",
"         ...,\n",
"         [0.43387375, 0.46650029, 0.27870709, 0.49094832],\n",
"         [0.68506131, 0.71279425, 0.52306111, 0.73242933],\n",
"         [0.93390885, 0.94159686, 0.87691377, 0.94675804]],\n",
"\n",
"        [[0.39898127, 0.6064991 , 0.35631382, 0.53576813],\n",
"         [0.96773129, 0.9858416 , 0.96154928, 0.98118091],\n",
"         [0.39923143, 0.60674802, 0.35655309, 0.53602756],\n",
"         ...,\n",
"         [0.27258513, 0.46525247, 0.23808017, 0.39448049],\n",
"         [0.5240085 , 0.71878432, 0.47861806, 0.65681593],\n",
"         [0.75235642, 0.87583334, 0.71698028, 0.8408075 ]],\n",
"\n",
"        ...,\n",
"\n",
"        [[0.41490246, 0.58450415, 0.37053763, 0.54497028],\n",
"         [0.98876773, 0.99430636, 0.98650025, 0.99331892],\n",
"         [0.43087656, 0.6003085 , 0.38593031, 0.56114936],\n",
"         ...,\n",
"         [0.43920209, 0.60840805, 0.39398858, 0.56947333],\n",
"         [0.6227391 , 0.76606381, 0.57810883, 0.73600259],\n",
"         [0.81660704, 0.89830698, 0.78706953, 0.88263577]],\n",
"\n",
"        [[0.32605625, 0.55958368, 0.33568331, 0.44382289],\n",
"         [0.99091996, 0.99652301, 0.991303  , 0.99447522],\n",
"         [0.33447933, 0.56894664, 0.34422806, 0.45324226],\n",
"         ...,\n",
"         [0.27791776, 0.50268386, 0.28672819, 0.38831555],\n",
"         [0.57631247, 0.78129058, 0.58689393, 0.69169794],\n",
"         [0.85558417, 0.93960975, 0.86087465, 0.90716521]],\n",
"\n",
"        [[0.43919597, 0.4573129 , 0.27618897, 0.51587419],\n",
"         [0.96495737, 0.96735202, 0.93063595, 0.97400374],\n",
"         [0.28225303, 0.29732839, 0.16079397, 0.34856129],\n",
"         ...,\n",
"         [0.48798296, 0.5062955 , 0.3171075 , 0.56460348],\n",
"         [0.62448024, 0.64149743, 0.44759056, 0.69350404],\n",
"         [0.88253997, 0.88992416, 0.7854454 , 0.91089806]]],\n",
"\n",
"\n",
"       [[[0.33882725, 0.57079158, 0.34207812, 0.45889441],\n",
"         [0.9670523 , 0.98704121, 0.96751049, 0.97982761],\n",
"         [0.32168238, 0.55170335, 0.32484958, 0.43971458],\n",
"         ...,\n",
"         [0.37471004, 0.60862671, 0.37810831, 0.49791689],\n",
"         [0.55778962, 0.76598949, 0.56135765, 0.67610448],\n",
"         [0.76175935, 0.89244434, 0.76437682, 0.84105246]],\n",
"\n",
"        [[0.46448215, 0.58500987, 0.29652301, 0.55451001],\n",
"         [0.9892782 , 0.99337582, 0.97818493, 0.99250442],\n",
"         [0.28011381, 0.38741005, 0.15902555, 0.35831708],\n",
"         ...,\n",
"         [0.41229425, 0.53275203, 0.25424698, 0.50168301],\n",
"         [0.48401999, 0.60390032, 0.31312689, 0.5737773 ],\n",
"         [0.9367197 , 0.96009364, 0.87795557, 0.9550422 ]],\n",
"\n",
"        [[0.43217251, 0.57116494, 0.28112784, 0.58953861],\n",
"         [0.98198767, 0.98962699, 0.96553177, 0.99037361],\n",
"         [0.242193  , 0.35868059, 0.14105258, 0.37621555],\n",
"         ...,\n",
"         [0.31428593, 0.44508267, 0.19061196, 0.46378693],\n",
"         [0.3850212 , 0.52281142, 0.24339225, 0.5415941 ],\n",
"         [0.93093208, 0.95932805, 0.87382549, 0.96217212]],\n",
"\n",
"        ...,\n",
"\n",
"        [[0.35989752, 0.4694215 , 0.28227482, 0.43124694],\n",
"         [0.95551066, 0.97126095, 0.93759073, 0.96662617],\n",
"         [0.39926292, 0.51119935, 0.31735935, 0.47265388],\n",
"         ...,\n",
"         [0.37896278, 0.48984868, 0.29914982, 0.4514263 ],\n",
"         [0.705035  , 0.78996805, 0.62574232, 0.7632235 ],\n",
"         [0.91593476, 0.94488775, 0.88400907, 0.93627877]],\n",
"\n",
"        [[0.24987052, 0.40970954, 0.22444823, 0.38041695],\n",
"         [0.98702358, 0.99373005, 0.98509341, 0.99291795],\n",
"         [0.26946694, 0.4345792 , 0.24269594, 0.40472766],\n",
"         ...,\n",
"         [0.36534672, 0.5453512 , 0.33339745, 0.51481869],\n",
"         [0.52204904, 0.69474347, 0.48690967, 0.66813845],\n",
"         [0.78400761, 0.88322307, 0.75924576, 0.8699707 ]],\n",
"\n",
"        [[0.38820285, 0.55787864, 0.3158555 , 0.49478664],\n",
"         [0.94240611, 0.97018426, 0.92251421, 0.96191253],\n",
"         [0.33664455, 0.50228659, 0.26967051, 0.43923563],\n",
"         ...,\n",
"         [0.37365979, 0.5426168 , 0.30268184, 0.47937935],\n",
"         [0.66183669, 0.79558406, 0.58746081, 0.7512908 ],\n",
"         [0.91187914, 0.95365674, 0.88275549, 0.94107829]]]])
• chain
(chain)
int64
0 1 2 3
array([0, 1, 2, 3])
• draw
(draw)
int64
0 1 2 3 4 5 ... 495 496 497 498 499
array([  0,   1,   2, ..., 497, 498, 499])
• actor
(actor)
int64
0 1 2 3 4 5 6
array([0, 1, 2, 3, 4, 5, 6])
• treatment
(treatment)
<U11
'treatment 1' ... 'treatment 4'
array(['treatment 1', 'treatment 2', 'treatment 3', 'treatment 4'],\n",
"      dtype='<U11')
" ], "text/plain": [ "\n", "array([[[[0.4441712 , 0.55402312, 0.3159088 , 0.45416202],\n", " [0.95606827, 0.97129007, 0.92634161, 0.95773349],\n", " [0.38020658, 0.48813174, 0.26171787, 0.38976755],\n", " ...,\n", " [0.42020547, 0.5297796 , 0.29518825, 0.43007432],\n", " [0.68420601, 0.77106923, 0.55595995, 0.69286569],\n", " [0.88616883, 0.92367647, 0.81814109, 0.89017923]],\n", "\n", " [[0.32700514, 0.46045575, 0.26880136, 0.50494431],\n", " [0.99531797, 0.99732888, 0.99382087, 0.99776409],\n", " [0.26036758, 0.38206201, 0.21031801, 0.42494175],\n", " ...,\n", " [0.33704464, 0.47172058, 0.27779151, 0.51625612],\n", " [0.58231819, 0.71003434, 0.51333344, 0.74532588],\n", " [0.90127602, 0.94129532, 0.87352952, 0.95040619]],\n", "\n", " [[0.25821549, 0.37166134, 0.25319262, 0.38688608],\n", " [0.90224481, 0.94005933, 0.89989218, 0.94360161],\n", " [0.3627958 , 0.49173028, 0.3567169 , 0.5078978 ],\n", " ...,\n", " [0.37010869, 0.49960439, 0.36397725, 0.51576809],\n", " [0.57781228, 0.69930008, 0.5713611 , 0.71272237],\n", " [0.79549626, 0.86858989, 0.7911692 , 0.87579774]],\n", "\n", " ...,\n", "\n", " [[0.41141284, 0.5076597 , 0.34369729, 0.61522334],\n", " [0.97425583, 0.98240233, 0.96593192, 0.98858014],\n", " [0.31092585, 0.39962558, 0.25265019, 0.50791326],\n", " ...,\n", " [0.37596846, 0.47055246, 0.31100422, 0.57950811],\n", " [0.54872531, 0.64205411, 0.47671404, 0.73555108],\n", " [0.82702274, 0.87582137, 0.78175772, 0.9162246 ]],\n", "\n", " [[0.34860267, 0.44440894, 0.25799757, 0.53637973],\n", " [0.98473185, 0.989733 , 0.97669224, 0.99287902],\n", " [0.3371221 , 0.431865 , 0.2483632 , 0.52368672],\n", " ...,\n", " [0.38896937, 0.48756585, 0.29258508, 0.5791578 ],\n", " [0.54462626, 0.64127009, 0.43727369, 0.72110408],\n", " [0.86547961, 0.90580591, 0.80695625, 0.93292613]],\n", "\n", " [[0.27939853, 0.56563795, 0.20965545, 0.493642 ],\n", " [0.95627397, 0.98656842, 0.93735302, 0.98213902],\n", " [0.20327621, 0.46147166, 0.14861582, 0.39080524],\n", " ...,\n", " [0.32762106, 0.62070895, 0.25001713, 0.55058863],\n", " [0.49769836, 0.76893658, 0.4040151 , 0.71357424],\n", " [0.90461133, 0.96955952, 0.86645702, 0.95974986]]],\n", "\n", "\n", " [[[0.29230878, 0.49718931, 0.24311653, 0.5228141 ],\n", " [0.9521296 , 0.97943043, 0.93927385, 0.98139822],\n", " [0.25408821, 0.44918316, 0.20942462, 0.47466948],\n", " ...,\n", " [0.24656325, 0.43928306, 0.20286266, 0.46468154],\n", " [0.59373307, 0.7777105 , 0.53194348, 0.79493551],\n", " [0.79660123, 0.9036226 , 0.75282097, 0.91219239]],\n", "\n", " [[0.36635053, 0.50596793, 0.30590282, 0.5609953 ],\n", " [0.9588454 , 0.9763434 , 0.9466954 , 0.98095088],\n", " [0.31199312, 0.44545801, 0.25687839, 0.5005725 ],\n", " ...,\n", " [0.28453008, 0.41330423, 0.23262644, 0.46779642],\n", " [0.61241248, 0.73676902, 0.54637264, 0.77739888],\n", " [0.85471835, 0.91244622, 0.81767268, 0.92858841]],\n", "\n", " [[0.3837187 , 0.61631435, 0.26100551, 0.55675367],\n", " [0.96468772, 0.98600969, 0.93938132, 0.98217845],\n", " [0.26720625, 0.48472571, 0.17139132, 0.42383437],\n", " ...,\n", " [0.29825557, 0.52301093, 0.19425843, 0.46161878],\n", " [0.75254097, 0.88694793, 0.63303408, 0.85984479],\n", " [0.85016375, 0.93605275, 0.7629517 , 0.91965561]],\n", "\n", " ...,\n", "\n", " [[0.41093853, 0.471524 , 0.23719419, 0.36942022],\n", " [0.97972067, 0.98407364, 0.9556224 , 0.97594463],\n", " [0.5141998 , 0.57514494, 0.3205549 , 0.47058315],\n", " ...,\n", " [0.33074744, 0.38728313, 0.18051772, 0.29329707],\n", " [0.78581351, 0.8243254 , 0.62053862, 0.75496207],\n", " [0.8424088 , 0.87239699, 0.70437595, 0.81781942]],\n", "\n", " [[0.27873963, 0.49098128, 0.31845388, 0.48612488],\n", " [0.98659982, 0.99458762, 0.98889104, 0.99448198],\n", " [0.2111228 , 0.40046514, 0.24446853, 0.39580787],\n", " ...,\n", " [0.3928927 , 0.61762373, 0.4389725 , 0.61302327],\n", " [0.44795244, 0.66944948, 0.49522232, 0.66513448],\n", " [0.82057904, 0.91945157, 0.84685072, 0.91800035]],\n", "\n", " [[0.32903585, 0.56608959, 0.37624354, 0.55591704],\n", " [0.96982435, 0.98843961, 0.97532802, 0.98795796],\n", " [0.2498199 , 0.46975951, 0.29058441, 0.45948493],\n", " ...,\n", " [0.30186593, 0.53495168, 0.34719259, 0.5246621 ],\n", " [0.51113316, 0.73555719, 0.56256182, 0.72744475],\n", " [0.85048785, 0.93801629, 0.87495055, 0.93557076]]],\n", "\n", "\n", " [[[0.3611322 , 0.46407287, 0.25932478, 0.44781883],\n", " [0.99303658, 0.99544332, 0.98880544, 0.99513622],\n", " [0.36113419, 0.46407502, 0.25932644, 0.44782097],\n", " ...,\n", " [0.35741229, 0.46005618, 0.25623296, 0.44382632],\n", " [0.59177783, 0.68950723, 0.47309906, 0.67530667],\n", " [0.81096574, 0.86793177, 0.72656619, 0.86023745]],\n", "\n", " [[0.42576834, 0.45827876, 0.27210715, 0.48268575],\n", " [0.94957421, 0.95552674, 0.90471008, 0.95950985],\n", " [0.32469497, 0.35424883, 0.19511682, 0.37697084],\n", " ...,\n", " [0.43387375, 0.46650029, 0.27870709, 0.49094832],\n", " [0.68506131, 0.71279425, 0.52306111, 0.73242933],\n", " [0.93390885, 0.94159686, 0.87691377, 0.94675804]],\n", "\n", " [[0.39898127, 0.6064991 , 0.35631382, 0.53576813],\n", " [0.96773129, 0.9858416 , 0.96154928, 0.98118091],\n", " [0.39923143, 0.60674802, 0.35655309, 0.53602756],\n", " ...,\n", " [0.27258513, 0.46525247, 0.23808017, 0.39448049],\n", " [0.5240085 , 0.71878432, 0.47861806, 0.65681593],\n", " [0.75235642, 0.87583334, 0.71698028, 0.8408075 ]],\n", "\n", " ...,\n", "\n", " [[0.41490246, 0.58450415, 0.37053763, 0.54497028],\n", " [0.98876773, 0.99430636, 0.98650025, 0.99331892],\n", " [0.43087656, 0.6003085 , 0.38593031, 0.56114936],\n", " ...,\n", " [0.43920209, 0.60840805, 0.39398858, 0.56947333],\n", " [0.6227391 , 0.76606381, 0.57810883, 0.73600259],\n", " [0.81660704, 0.89830698, 0.78706953, 0.88263577]],\n", "\n", " [[0.32605625, 0.55958368, 0.33568331, 0.44382289],\n", " [0.99091996, 0.99652301, 0.991303 , 0.99447522],\n", " [0.33447933, 0.56894664, 0.34422806, 0.45324226],\n", " ...,\n", " [0.27791776, 0.50268386, 0.28672819, 0.38831555],\n", " [0.57631247, 0.78129058, 0.58689393, 0.69169794],\n", " [0.85558417, 0.93960975, 0.86087465, 0.90716521]],\n", "\n", " [[0.43919597, 0.4573129 , 0.27618897, 0.51587419],\n", " [0.96495737, 0.96735202, 0.93063595, 0.97400374],\n", " [0.28225303, 0.29732839, 0.16079397, 0.34856129],\n", " ...,\n", " [0.48798296, 0.5062955 , 0.3171075 , 0.56460348],\n", " [0.62448024, 0.64149743, 0.44759056, 0.69350404],\n", " [0.88253997, 0.88992416, 0.7854454 , 0.91089806]]],\n", "\n", "\n", " [[[0.33882725, 0.57079158, 0.34207812, 0.45889441],\n", " [0.9670523 , 0.98704121, 0.96751049, 0.97982761],\n", " [0.32168238, 0.55170335, 0.32484958, 0.43971458],\n", " ...,\n", " [0.37471004, 0.60862671, 0.37810831, 0.49791689],\n", " [0.55778962, 0.76598949, 0.56135765, 0.67610448],\n", " [0.76175935, 0.89244434, 0.76437682, 0.84105246]],\n", "\n", " [[0.46448215, 0.58500987, 0.29652301, 0.55451001],\n", " [0.9892782 , 0.99337582, 0.97818493, 0.99250442],\n", " [0.28011381, 0.38741005, 0.15902555, 0.35831708],\n", " ...,\n", " [0.41229425, 0.53275203, 0.25424698, 0.50168301],\n", " [0.48401999, 0.60390032, 0.31312689, 0.5737773 ],\n", " [0.9367197 , 0.96009364, 0.87795557, 0.9550422 ]],\n", "\n", " [[0.43217251, 0.57116494, 0.28112784, 0.58953861],\n", " [0.98198767, 0.98962699, 0.96553177, 0.99037361],\n", " [0.242193 , 0.35868059, 0.14105258, 0.37621555],\n", " ...,\n", " [0.31428593, 0.44508267, 0.19061196, 0.46378693],\n", " [0.3850212 , 0.52281142, 0.24339225, 0.5415941 ],\n", " [0.93093208, 0.95932805, 0.87382549, 0.96217212]],\n", "\n", " ...,\n", "\n", " [[0.35989752, 0.4694215 , 0.28227482, 0.43124694],\n", " [0.95551066, 0.97126095, 0.93759073, 0.96662617],\n", " [0.39926292, 0.51119935, 0.31735935, 0.47265388],\n", " ...,\n", " [0.37896278, 0.48984868, 0.29914982, 0.4514263 ],\n", " [0.705035 , 0.78996805, 0.62574232, 0.7632235 ],\n", " [0.91593476, 0.94488775, 0.88400907, 0.93627877]],\n", "\n", " [[0.24987052, 0.40970954, 0.22444823, 0.38041695],\n", " [0.98702358, 0.99373005, 0.98509341, 0.99291795],\n", " [0.26946694, 0.4345792 , 0.24269594, 0.40472766],\n", " ...,\n", " [0.36534672, 0.5453512 , 0.33339745, 0.51481869],\n", " [0.52204904, 0.69474347, 0.48690967, 0.66813845],\n", " [0.78400761, 0.88322307, 0.75924576, 0.8699707 ]],\n", "\n", " [[0.38820285, 0.55787864, 0.3158555 , 0.49478664],\n", " [0.94240611, 0.97018426, 0.92251421, 0.96191253],\n", " [0.33664455, 0.50228659, 0.26967051, 0.43923563],\n", " ...,\n", " [0.37365979, 0.5426168 , 0.30268184, 0.47937935],\n", " [0.66183669, 0.79558406, 0.58746081, 0.7512908 ],\n", " [0.91187914, 0.95365674, 0.88275549, 0.94107829]]]])\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", " * actor (actor) int64 0 1 2 3 4 5 6\n", " * treatment (treatment) \n", "
xarray.DataArray
• actor: 7
• treatment: 4
• 0.3827 0.5079 0.3065 0.4803 0.9738 ... 0.8651 0.9141 0.8205 0.9051
array([[0.38273298, 0.50794882, 0.3064966 , 0.48028027],\n",
"       [0.97383758, 0.9842027 , 0.96366044, 0.98230814],\n",
"       [0.31552197, 0.43399235, 0.24723306, 0.40704301],\n",
"       [0.31658471, 0.43514105, 0.24811081, 0.40830217],\n",
"       [0.38371248, 0.50898154, 0.30735674, 0.4813107 ],\n",
"       [0.60644722, 0.71919498, 0.52349316, 0.69665124],\n",
"       [0.8650691 , 0.91412716, 0.82050517, 0.90508358]])
• actor
(actor)
int64
0 1 2 3 4 5 6
array([0, 1, 2, 3, 4, 5, 6])
• treatment
(treatment)
<U11
'treatment 1' ... 'treatment 4'
array(['treatment 1', 'treatment 2', 'treatment 3', 'treatment 4'],\n",
"      dtype='<U11')
" ], "text/plain": [ "\n", "array([[0.38273298, 0.50794882, 0.3064966 , 0.48028027],\n", " [0.97383758, 0.9842027 , 0.96366044, 0.98230814],\n", " [0.31552197, 0.43399235, 0.24723306, 0.40704301],\n", " [0.31658471, 0.43514105, 0.24811081, 0.40830217],\n", " [0.38371248, 0.50898154, 0.30735674, 0.4813107 ],\n", " [0.60644722, 0.71919498, 0.52349316, 0.69665124],\n", " [0.8650691 , 0.91412716, 0.82050517, 0.90508358]])\n", "Coordinates:\n", " * actor (actor) int64 0 1 2 3 4 5 6\n", " * treatment (treatment) Am I correct thinking that since p is a deterministic function of the parameters a and b , there should be a value of p in trace_11_4 for every value returned by sample_posterior_predictive?\n", "\n", "We will explore the values stored in p in trace_11_4 (or what is the same, in idata_11_4). p is defined in the following line:\n", "\n", " p = pm.Deterministic(\"p\", pm.math.invlogit(a[actor_id] + b[treat_id]))\n", "\n", "Thus, at every sample, p will contain only 28 different values, these unique values will be equal to the values returned by pm.sample_posterior_predictive and will be equal to the values stored in p_mu_samples. To check this, we'll compare the values of p in each sample for a couple of actor-treatment combination with the corresponding values in the other alternatives. " ] }, { "cell_type": "code", "execution_count": 15, "metadata": {}, "outputs": [], "source": [ "def get_mask(actor, treatment):\n", " return (actor == actor_idx) & (treatment == treat_idx)" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "Example 1: actor 2 and treatment 1" ] }, { "cell_type": "code", "execution_count": 16, "metadata": {}, "outputs": [], "source": [ "mask_2_1 = get_mask(actor=2, treatment=1)" ] }, { "cell_type": "code", "execution_count": 17, "metadata": {}, "outputs": [ { "data": { "text/html": [ "
\n", "
xarray.DataArray
'p'
• chain: 4
• draw: 500
• experiment: 18
• 0.4881 0.4881 0.4881 0.4881 0.4881 ... 0.5023 0.5023 0.5023 0.5023
array([[[0.48813174, 0.48813174, 0.48813174, ..., 0.48813174,\n",
"         0.48813174, 0.48813174],\n",
"        [0.38206201, 0.38206201, 0.38206201, ..., 0.38206201,\n",
"         0.38206201, 0.38206201],\n",
"        [0.49173028, 0.49173028, 0.49173028, ..., 0.49173028,\n",
"         0.49173028, 0.49173028],\n",
"        ...,\n",
"        [0.39962558, 0.39962558, 0.39962558, ..., 0.39962558,\n",
"         0.39962558, 0.39962558],\n",
"        [0.431865  , 0.431865  , 0.431865  , ..., 0.431865  ,\n",
"         0.431865  , 0.431865  ],\n",
"        [0.46147166, 0.46147166, 0.46147166, ..., 0.46147166,\n",
"         0.46147166, 0.46147166]],\n",
"\n",
"       [[0.44918316, 0.44918316, 0.44918316, ..., 0.44918316,\n",
"         0.44918316, 0.44918316],\n",
"        [0.44545801, 0.44545801, 0.44545801, ..., 0.44545801,\n",
"         0.44545801, 0.44545801],\n",
"        [0.48472571, 0.48472571, 0.48472571, ..., 0.48472571,\n",
"         0.48472571, 0.48472571],\n",
"        ...,\n",
"        [0.57514494, 0.57514494, 0.57514494, ..., 0.57514494,\n",
"         0.57514494, 0.57514494],\n",
"        [0.40046514, 0.40046514, 0.40046514, ..., 0.40046514,\n",
"         0.40046514, 0.40046514],\n",
"        [0.46975951, 0.46975951, 0.46975951, ..., 0.46975951,\n",
"         0.46975951, 0.46975951]],\n",
"\n",
"       [[0.46407502, 0.46407502, 0.46407502, ..., 0.46407502,\n",
"         0.46407502, 0.46407502],\n",
"        [0.35424883, 0.35424883, 0.35424883, ..., 0.35424883,\n",
"         0.35424883, 0.35424883],\n",
"        [0.60674802, 0.60674802, 0.60674802, ..., 0.60674802,\n",
"         0.60674802, 0.60674802],\n",
"        ...,\n",
"        [0.6003085 , 0.6003085 , 0.6003085 , ..., 0.6003085 ,\n",
"         0.6003085 , 0.6003085 ],\n",
"        [0.56894664, 0.56894664, 0.56894664, ..., 0.56894664,\n",
"         0.56894664, 0.56894664],\n",
"        [0.29732839, 0.29732839, 0.29732839, ..., 0.29732839,\n",
"         0.29732839, 0.29732839]],\n",
"\n",
"       [[0.55170335, 0.55170335, 0.55170335, ..., 0.55170335,\n",
"         0.55170335, 0.55170335],\n",
"        [0.38741005, 0.38741005, 0.38741005, ..., 0.38741005,\n",
"         0.38741005, 0.38741005],\n",
"        [0.35868059, 0.35868059, 0.35868059, ..., 0.35868059,\n",
"         0.35868059, 0.35868059],\n",
"        ...,\n",
"        [0.51119935, 0.51119935, 0.51119935, ..., 0.51119935,\n",
"         0.51119935, 0.51119935],\n",
"        [0.4345792 , 0.4345792 , 0.4345792 , ..., 0.4345792 ,\n",
"         0.4345792 , 0.4345792 ],\n",
"        [0.50228659, 0.50228659, 0.50228659, ..., 0.50228659,\n",
"         0.50228659, 0.50228659]]])
• chain
(chain)
int64
0 1 2 3
array([0, 1, 2, 3])
• draw
(draw)
int64
0 1 2 3 4 5 ... 495 496 497 498 499
array([  0,   1,   2, ..., 497, 498, 499])
• experiment
(experiment)
int64
145 146 147 150 ... 173 176 177 179
array([145, 146, 147, 150, 152, 153, 156, 160, 161, 162, 166, 167, 169, 170,\n",
"       173, 176, 177, 179])
" ], "text/plain": [ "\n", "array([[[0.48813174, 0.48813174, 0.48813174, ..., 0.48813174,\n", " 0.48813174, 0.48813174],\n", " [0.38206201, 0.38206201, 0.38206201, ..., 0.38206201,\n", " 0.38206201, 0.38206201],\n", " [0.49173028, 0.49173028, 0.49173028, ..., 0.49173028,\n", " 0.49173028, 0.49173028],\n", " ...,\n", " [0.39962558, 0.39962558, 0.39962558, ..., 0.39962558,\n", " 0.39962558, 0.39962558],\n", " [0.431865 , 0.431865 , 0.431865 , ..., 0.431865 ,\n", " 0.431865 , 0.431865 ],\n", " [0.46147166, 0.46147166, 0.46147166, ..., 0.46147166,\n", " 0.46147166, 0.46147166]],\n", "\n", " [[0.44918316, 0.44918316, 0.44918316, ..., 0.44918316,\n", " 0.44918316, 0.44918316],\n", " [0.44545801, 0.44545801, 0.44545801, ..., 0.44545801,\n", " 0.44545801, 0.44545801],\n", " [0.48472571, 0.48472571, 0.48472571, ..., 0.48472571,\n", " 0.48472571, 0.48472571],\n", " ...,\n", " [0.57514494, 0.57514494, 0.57514494, ..., 0.57514494,\n", " 0.57514494, 0.57514494],\n", " [0.40046514, 0.40046514, 0.40046514, ..., 0.40046514,\n", " 0.40046514, 0.40046514],\n", " [0.46975951, 0.46975951, 0.46975951, ..., 0.46975951,\n", " 0.46975951, 0.46975951]],\n", "\n", " [[0.46407502, 0.46407502, 0.46407502, ..., 0.46407502,\n", " 0.46407502, 0.46407502],\n", " [0.35424883, 0.35424883, 0.35424883, ..., 0.35424883,\n", " 0.35424883, 0.35424883],\n", " [0.60674802, 0.60674802, 0.60674802, ..., 0.60674802,\n", " 0.60674802, 0.60674802],\n", " ...,\n", " [0.6003085 , 0.6003085 , 0.6003085 , ..., 0.6003085 ,\n", " 0.6003085 , 0.6003085 ],\n", " [0.56894664, 0.56894664, 0.56894664, ..., 0.56894664,\n", " 0.56894664, 0.56894664],\n", " [0.29732839, 0.29732839, 0.29732839, ..., 0.29732839,\n", " 0.29732839, 0.29732839]],\n", "\n", " [[0.55170335, 0.55170335, 0.55170335, ..., 0.55170335,\n", " 0.55170335, 0.55170335],\n", " [0.38741005, 0.38741005, 0.38741005, ..., 0.38741005,\n", " 0.38741005, 0.38741005],\n", " [0.35868059, 0.35868059, 0.35868059, ..., 0.35868059,\n", " 0.35868059, 0.35868059],\n", " ...,\n", " [0.51119935, 0.51119935, 0.51119935, ..., 0.51119935,\n", " 0.51119935, 0.51119935],\n", " [0.4345792 , 0.4345792 , 0.4345792 , ..., 0.4345792 ,\n", " 0.4345792 , 0.4345792 ],\n", " [0.50228659, 0.50228659, 0.50228659, ..., 0.50228659,\n", " 0.50228659, 0.50228659]]])\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", " * experiment (experiment) int64 145 146 147 150 152 ... 170 173 176 177 179" ] }, "execution_count": 17, "metadata": {}, "output_type": "execute_result" } ], "source": [ "da_2_1 = idata_11_4.posterior.p.sel(experiment=mask_2_1) \n", "da_2_1 # the database icon can be used to explore the data" ] }, { "cell_type": "code", "execution_count": 18, "metadata": {}, "outputs": [ { "data": { "text/plain": [ "True" ] }, "execution_count": 18, "metadata": {}, "output_type": "execute_result" } ], "source": [ "np.allclose(da_2_1.values, p_post.reshape((4,-1,7,4))[:,:,2,1,None])" ] }, { "cell_type": "code", "execution_count": 19, "metadata": {}, "outputs": [ { "data": { "text/plain": [ "True" ] }, "execution_count": 19, "metadata": {}, "output_type": "execute_result" } ], "source": [ "np.allclose(da_2_1.values, p_mu_samples.isel(actor=2, treatment=1).values[:,:,None])" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "_As shown in this example, positional indexing, isel, can still be used with xarray. Below we use label based indexing, sel_" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "Example 1: actor 5 and treatment 3" ] }, { "cell_type": "code", "execution_count": 20, "metadata": {}, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "True\n", "True\n" ] } ], "source": [ "mask_5_3 = get_mask(actor=5, treatment=3)\n", "da_5_3 = idata_11_4.posterior.p.sel(experiment=mask_5_3) \n", "print(np.allclose(da_5_3.values, p_post.reshape((4,-1,7,4))[:,:,5,3,None]))\n", "print(np.allclose(da_5_3.values, p_mu_samples.sel(actor=5, treatment=\"treatment 4\").values[:,:,None]))" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "## A mixed pandas-xarray implementation of option 1" ] }, { "cell_type": "code", "execution_count": 21, "metadata": {}, "outputs": [ { "data": { "text/html": [ "
\n", "\n", "\n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", "
actorrecipientconditionblocktrialprosoc_leftchose_prosocpulled_lefttreatmentrepetition
00NaN01201000
10NaN01400101
20NaN01610010
30NaN01801002
40NaN011011111
.................................
49964.01664111316
50066.01666111317
50163.01668001215
50267.01670001216
50362.01672001217
\n", "

504 rows × 10 columns

\n", "
" ], "text/plain": [ " actor recipient condition block trial prosoc_left chose_prosoc \\\n", "0 0 NaN 0 1 2 0 1 \n", "1 0 NaN 0 1 4 0 0 \n", "2 0 NaN 0 1 6 1 0 \n", "3 0 NaN 0 1 8 0 1 \n", "4 0 NaN 0 1 10 1 1 \n", ".. ... ... ... ... ... ... ... \n", "499 6 4.0 1 6 64 1 1 \n", "500 6 6.0 1 6 66 1 1 \n", "501 6 3.0 1 6 68 0 0 \n", "502 6 7.0 1 6 70 0 0 \n", "503 6 2.0 1 6 72 0 0 \n", "\n", " pulled_left treatment repetition \n", "0 0 0 0 \n", "1 1 0 1 \n", "2 0 1 0 \n", "3 0 0 2 \n", "4 1 1 1 \n", ".. ... ... ... \n", "499 1 3 16 \n", "500 1 3 17 \n", "501 1 2 15 \n", "502 1 2 16 \n", "503 1 2 17 \n", "\n", "[504 rows x 10 columns]" ] }, "execution_count": 21, "metadata": {}, "output_type": "execute_result" } ], "source": [ "d[\"repetition\"] = d.groupby([\"actor\", \"treatment\"]).cumcount()\n", "d" ] }, { "cell_type": "code", "execution_count": 22, "metadata": {}, "outputs": [ { "data": { "text/html": [ "
\n", "
xarray.Dataset
• actor: 7
• chain: 4
• draw: 500
• repetition: 18
• treatment: 4
• chain
(chain)
int64
0 1 2 3
array([0, 1, 2, 3])
• draw
(draw)
int64
0 1 2 3 4 5 ... 495 496 497 498 499
array([  0,   1,   2, ..., 497, 498, 499])
• actor
(actor)
int64
0 1 2 3 4 5 6
array([0, 1, 2, 3, 4, 5, 6])
• treatment
(treatment)
<U11
'treatment 1' ... 'treatment 4'
array(['treatment 1', 'treatment 2', 'treatment 3', 'treatment 4'],\n",
"      dtype='<U11')
• repetition
(repetition)
int64
0 1 2 3 4 5 6 ... 12 13 14 15 16 17
array([ 0,  1,  2,  3,  4,  5,  6,  7,  8,  9, 10, 11, 12, 13, 14, 15, 16, 17])
• a
(chain, draw, actor)
float64
0.135 3.439 ... 0.8038 2.469
array([[[ 0.1350494 ,  3.43949205, -0.12937185, ...,  0.03736965,\n",
"          1.13246861,  2.41149076],\n",
"        [-0.53867798,  5.54241401, -0.86097509, ..., -0.49340846,\n",
"          0.51538093,  2.39456749],\n",
"        [-0.6464498 ,  2.63123411, -0.15443548, ..., -0.12293604,\n",
"          0.72261353,  1.76719445],\n",
"        ...,\n",
"        [-0.22109826,  3.77049511, -0.65876488, ..., -0.36966607,\n",
"          0.3325513 ,  1.70170163],\n",
"        [-0.46268774,  4.32909982, -0.51364639, ..., -0.28914722,\n",
"          0.34148056,  2.02406726],\n",
"        [-0.79424648,  3.23830151, -1.2127418 , ..., -0.56576385,\n",
"          0.14399398,  2.40274617]],\n",
"\n",
"       [[-0.44553248,  3.42886858, -0.63826118, ..., -0.67836183,\n",
"          0.81808415,  1.80385049],\n",
"        [-0.28375707,  3.41254227, -0.52666946, ..., -0.65795236,\n",
"          0.72161261,  2.03624599],\n",
"        [-0.24907328,  3.5322941 , -0.78412297, ..., -0.63089803,\n",
"          1.33693092,  1.96060643],\n",
"        ...,\n",
"        [-0.15288734,  4.08486494,  0.26401406, ..., -0.49760689,\n",
"          1.507072  ,  1.88346066],\n",
"        [-0.42494098,  4.82477725, -0.79238959, ...,  0.09061211,\n",
"          0.31683397,  2.04605659],\n",
"        [ 0.46118619,  4.64381499,  0.07416212, ...,  0.33530701,\n",
"          1.21827518,  2.91216765]],\n",
"\n",
"       [[-0.78547128,  4.7450789 , -0.78546263, ..., -0.80163116,\n",
"          0.15630192,  1.24127972],\n",
"        [-0.33331993,  2.90132854, -0.76646069, ..., -0.30024591,\n",
"          0.74294793,  2.61416152],\n",
"        [-0.39723684,  3.41333113, -0.39619374, ..., -0.96907126,\n",
"          0.10858268,  1.1236944 ],\n",
"        ...,\n",
"        [ 0.05353087,  4.87493456,  0.11899031, ...,  0.15286503,\n",
"          0.89845667,  1.89079282],\n",
"        [-0.17776394,  5.24086845, -0.13968179, ..., -0.40650101,\n",
"          0.85596658,  2.32740041],\n",
"        [-0.06807385,  3.49187068, -0.7569612 , ...,  0.12827456,\n",
"          0.68496056,  2.1930579 ]],\n",
"\n",
"       [[-0.72787785,  3.3199782 , -0.80540404, ..., -0.57141605,\n",
"          0.17284305,  1.1029963 ],\n",
"        [-0.44173558,  4.22527233, -1.24332166, ..., -0.65391359,\n",
"         -0.36336629,  2.39538558],\n",
"        [-0.3523625 ,  3.91915261, -1.2200635 , ..., -0.85952723,\n",
"         -0.54765908,  2.52172622],\n",
"        ...,\n",
"        [-0.5660457 ,  3.07675958, -0.39877395, ..., -0.48418973,\n",
"          0.88115401,  2.39811526],\n",
"        [-0.57515718,  4.85570589, -0.47318295, ..., -0.02808618,\n",
"          0.61239922,  1.81332135],\n",
"        [-0.32259427,  2.92729824, -0.54600485, ..., -0.38426948,\n",
"          0.80376838,  2.46907673]]])
• b
(chain, draw, treatment)
float64
-0.3593 0.08189 ... -0.4503 0.3017
array([[[-0.35929964,  0.08188989, -0.90768709, -0.31891759],\n",
"        [-0.18308382,  0.38016992, -0.46203453,  0.55845587],\n",
"        [-0.40881454,  0.12135358, -0.43520694,  0.18602932],\n",
"        ...,\n",
"        [-0.13702956,  0.25173945, -0.42576259,  0.69042055],\n",
"        [-0.16249926,  0.23940038, -0.59371464,  0.60846425],\n",
"        [-0.15320059,  1.0583223 , -0.53275706,  0.76881312]],\n",
"\n",
"       [[-0.4386647 ,  0.43428959, -0.69013596,  0.5368523 ],\n",
"        [-0.26414823,  0.30762992, -0.53558744,  0.52895948],\n",
"        [-0.22472049,  0.7230068 , -0.79167565,  0.47707052],\n",
"        ...,\n",
"        [-0.20719957,  0.03885995, -1.01523696, -0.38181753],\n",
"        [-0.52578112,  0.3888622 , -0.33594527,  0.36942624],\n",
"        [-1.17373519, -0.19527191, -0.96670953, -0.23657849]],\n",
"\n",
"       [[ 0.21501781,  0.64151466, -0.2640097 ,  0.57598384],\n",
"        [ 0.0341824 ,  0.16604603, -0.65063801,  0.26403522],\n",
"        [-0.01247477,  0.82985674, -0.19416263,  0.54055417],\n",
"        ...,\n",
"        [-0.397266  ,  0.2877604 , -0.58344193,  0.12683765],\n",
"        [-0.5483128 ,  0.41723657, -0.50482686, -0.04789725],\n",
"        [-0.17635197, -0.10309123, -0.8953712 ,  0.13159195]],\n",
"\n",
"       [[ 0.05935307,  1.01295935,  0.07383078,  0.56308356],\n",
"        [ 0.29942449,  0.78510956, -0.42217475,  0.66064565],\n",
"        [ 0.07936967,  0.63896814, -0.58651151,  0.71442086],\n",
"        ...,\n",
"        [-0.00976328,  0.44357886, -0.36715989,  0.28928022],\n",
"        [-0.52414581,  0.20999093, -0.66477251,  0.08737833],\n",
"        [-0.1322785 ,  0.55515128, -0.45029006,  0.30174006]]])
• p
(chain, draw, actor, treatment, repetition)
float64
0.4442 0.4442 ... 0.9411 0.9411
array([[[[[0.4441712 , 0.4441712 , 0.4441712 , ..., 0.4441712 ,\n",
"           0.4441712 , 0.4441712 ],\n",
"          [0.55402312, 0.55402312, 0.55402312, ..., 0.55402312,\n",
"           0.55402312, 0.55402312],\n",
"          [0.3159088 , 0.3159088 , 0.3159088 , ..., 0.3159088 ,\n",
"           0.3159088 , 0.3159088 ],\n",
"          [0.45416202, 0.45416202, 0.45416202, ..., 0.45416202,\n",
"           0.45416202, 0.45416202]],\n",
"\n",
"         [[0.95606827, 0.95606827, 0.95606827, ..., 0.95606827,\n",
"           0.95606827, 0.95606827],\n",
"          [0.97129007, 0.97129007, 0.97129007, ..., 0.97129007,\n",
"           0.97129007, 0.97129007],\n",
"          [0.92634161, 0.92634161, 0.92634161, ..., 0.92634161,\n",
"           0.92634161, 0.92634161],\n",
"          [0.95773349, 0.95773349, 0.95773349, ..., 0.95773349,\n",
"           0.95773349, 0.95773349]],\n",
"\n",
"         [[0.38020658, 0.38020658, 0.38020658, ..., 0.38020658,\n",
"           0.38020658, 0.38020658],\n",
"          [0.48813174, 0.48813174, 0.48813174, ..., 0.48813174,\n",
"           0.48813174, 0.48813174],\n",
"          [0.26171787, 0.26171787, 0.26171787, ..., 0.26171787,\n",
"           0.26171787, 0.26171787],\n",
"          [0.38976755, 0.38976755, 0.38976755, ..., 0.38976755,\n",
"           0.38976755, 0.38976755]],\n",
"\n",
"         ...,\n",
"\n",
"         [[0.42020547, 0.42020547, 0.42020547, ..., 0.42020547,\n",
"           0.42020547, 0.42020547],\n",
"          [0.5297796 , 0.5297796 , 0.5297796 , ..., 0.5297796 ,\n",
"           0.5297796 , 0.5297796 ],\n",
"          [0.29518825, 0.29518825, 0.29518825, ..., 0.29518825,\n",
"           0.29518825, 0.29518825],\n",
"          [0.43007432, 0.43007432, 0.43007432, ..., 0.43007432,\n",
"           0.43007432, 0.43007432]],\n",
"\n",
"         [[0.68420601, 0.68420601, 0.68420601, ..., 0.68420601,\n",
"           0.68420601, 0.68420601],\n",
"          [0.77106923, 0.77106923, 0.77106923, ..., 0.77106923,\n",
"           0.77106923, 0.77106923],\n",
"          [0.55595995, 0.55595995, 0.55595995, ..., 0.55595995,\n",
"           0.55595995, 0.55595995],\n",
"          [0.69286569, 0.69286569, 0.69286569, ..., 0.69286569,\n",
"           0.69286569, 0.69286569]],\n",
"\n",
"         [[0.88616883, 0.88616883, 0.88616883, ..., 0.88616883,\n",
"           0.88616883, 0.88616883],\n",
"          [0.92367647, 0.92367647, 0.92367647, ..., 0.92367647,\n",
"           0.92367647, 0.92367647],\n",
"          [0.81814109, 0.81814109, 0.81814109, ..., 0.81814109,\n",
"           0.81814109, 0.81814109],\n",
"          [0.89017923, 0.89017923, 0.89017923, ..., 0.89017923,\n",
"           0.89017923, 0.89017923]]],\n",
"\n",
"\n",
"        [[[0.32700514, 0.32700514, 0.32700514, ..., 0.32700514,\n",
"           0.32700514, 0.32700514],\n",
"          [0.46045575, 0.46045575, 0.46045575, ..., 0.46045575,\n",
"           0.46045575, 0.46045575],\n",
"          [0.26880136, 0.26880136, 0.26880136, ..., 0.26880136,\n",
"           0.26880136, 0.26880136],\n",
"          [0.50494431, 0.50494431, 0.50494431, ..., 0.50494431,\n",
"           0.50494431, 0.50494431]],\n",
"\n",
"         [[0.99531797, 0.99531797, 0.99531797, ..., 0.99531797,\n",
"           0.99531797, 0.99531797],\n",
"          [0.99732888, 0.99732888, 0.99732888, ..., 0.99732888,\n",
"           0.99732888, 0.99732888],\n",
"          [0.99382087, 0.99382087, 0.99382087, ..., 0.99382087,\n",
"           0.99382087, 0.99382087],\n",
"          [0.99776409, 0.99776409, 0.99776409, ..., 0.99776409,\n",
"           0.99776409, 0.99776409]],\n",
"\n",
"         [[0.26036758, 0.26036758, 0.26036758, ..., 0.26036758,\n",
"           0.26036758, 0.26036758],\n",
"          [0.38206201, 0.38206201, 0.38206201, ..., 0.38206201,\n",
"           0.38206201, 0.38206201],\n",
"          [0.21031801, 0.21031801, 0.21031801, ..., 0.21031801,\n",
"           0.21031801, 0.21031801],\n",
"          [0.42494175, 0.42494175, 0.42494175, ..., 0.42494175,\n",
"           0.42494175, 0.42494175]],\n",
"\n",
"         ...,\n",
"\n",
"         [[0.33704464, 0.33704464, 0.33704464, ..., 0.33704464,\n",
"           0.33704464, 0.33704464],\n",
"          [0.47172058, 0.47172058, 0.47172058, ..., 0.47172058,\n",
"           0.47172058, 0.47172058],\n",
"          [0.27779151, 0.27779151, 0.27779151, ..., 0.27779151,\n",
"           0.27779151, 0.27779151],\n",
"          [0.51625612, 0.51625612, 0.51625612, ..., 0.51625612,\n",
"           0.51625612, 0.51625612]],\n",
"\n",
"         [[0.58231819, 0.58231819, 0.58231819, ..., 0.58231819,\n",
"           0.58231819, 0.58231819],\n",
"          [0.71003434, 0.71003434, 0.71003434, ..., 0.71003434,\n",
"           0.71003434, 0.71003434],\n",
"          [0.51333344, 0.51333344, 0.51333344, ..., 0.51333344,\n",
"           0.51333344, 0.51333344],\n",
"          [0.74532588, 0.74532588, 0.74532588, ..., 0.74532588,\n",
"           0.74532588, 0.74532588]],\n",
"\n",
"         [[0.90127602, 0.90127602, 0.90127602, ..., 0.90127602,\n",
"           0.90127602, 0.90127602],\n",
"          [0.94129532, 0.94129532, 0.94129532, ..., 0.94129532,\n",
"           0.94129532, 0.94129532],\n",
"          [0.87352952, 0.87352952, 0.87352952, ..., 0.87352952,\n",
"           0.87352952, 0.87352952],\n",
"          [0.95040619, 0.95040619, 0.95040619, ..., 0.95040619,\n",
"           0.95040619, 0.95040619]]],\n",
"\n",
"\n",
"        [[[0.25821549, 0.25821549, 0.25821549, ..., 0.25821549,\n",
"           0.25821549, 0.25821549],\n",
"          [0.37166134, 0.37166134, 0.37166134, ..., 0.37166134,\n",
"           0.37166134, 0.37166134],\n",
"          [0.25319262, 0.25319262, 0.25319262, ..., 0.25319262,\n",
"           0.25319262, 0.25319262],\n",
"          [0.38688608, 0.38688608, 0.38688608, ..., 0.38688608,\n",
"           0.38688608, 0.38688608]],\n",
"\n",
"         [[0.90224481, 0.90224481, 0.90224481, ..., 0.90224481,\n",
"           0.90224481, 0.90224481],\n",
"          [0.94005933, 0.94005933, 0.94005933, ..., 0.94005933,\n",
"           0.94005933, 0.94005933],\n",
"          [0.89989218, 0.89989218, 0.89989218, ..., 0.89989218,\n",
"           0.89989218, 0.89989218],\n",
"          [0.94360161, 0.94360161, 0.94360161, ..., 0.94360161,\n",
"           0.94360161, 0.94360161]],\n",
"\n",
"         [[0.3627958 , 0.3627958 , 0.3627958 , ..., 0.3627958 ,\n",
"           0.3627958 , 0.3627958 ],\n",
"          [0.49173028, 0.49173028, 0.49173028, ..., 0.49173028,\n",
"           0.49173028, 0.49173028],\n",
"          [0.3567169 , 0.3567169 , 0.3567169 , ..., 0.3567169 ,\n",
"           0.3567169 , 0.3567169 ],\n",
"          [0.5078978 , 0.5078978 , 0.5078978 , ..., 0.5078978 ,\n",
"           0.5078978 , 0.5078978 ]],\n",
"\n",
"         ...,\n",
"\n",
"         [[0.37010869, 0.37010869, 0.37010869, ..., 0.37010869,\n",
"           0.37010869, 0.37010869],\n",
"          [0.49960439, 0.49960439, 0.49960439, ..., 0.49960439,\n",
"           0.49960439, 0.49960439],\n",
"          [0.36397725, 0.36397725, 0.36397725, ..., 0.36397725,\n",
"           0.36397725, 0.36397725],\n",
"          [0.51576809, 0.51576809, 0.51576809, ..., 0.51576809,\n",
"           0.51576809, 0.51576809]],\n",
"\n",
"         [[0.57781228, 0.57781228, 0.57781228, ..., 0.57781228,\n",
"           0.57781228, 0.57781228],\n",
"          [0.69930008, 0.69930008, 0.69930008, ..., 0.69930008,\n",
"           0.69930008, 0.69930008],\n",
"          [0.5713611 , 0.5713611 , 0.5713611 , ..., 0.5713611 ,\n",
"           0.5713611 , 0.5713611 ],\n",
"          [0.71272237, 0.71272237, 0.71272237, ..., 0.71272237,\n",
"           0.71272237, 0.71272237]],\n",
"\n",
"         [[0.79549626, 0.79549626, 0.79549626, ..., 0.79549626,\n",
"           0.79549626, 0.79549626],\n",
"          [0.86858989, 0.86858989, 0.86858989, ..., 0.86858989,\n",
"           0.86858989, 0.86858989],\n",
"          [0.7911692 , 0.7911692 , 0.7911692 , ..., 0.7911692 ,\n",
"           0.7911692 , 0.7911692 ],\n",
"          [0.87579774, 0.87579774, 0.87579774, ..., 0.87579774,\n",
"           0.87579774, 0.87579774]]],\n",
"\n",
"\n",
"        ...,\n",
"\n",
"\n",
"        [[[0.41141284, 0.41141284, 0.41141284, ..., 0.41141284,\n",
"           0.41141284, 0.41141284],\n",
"          [0.5076597 , 0.5076597 , 0.5076597 , ..., 0.5076597 ,\n",
"           0.5076597 , 0.5076597 ],\n",
"          [0.34369729, 0.34369729, 0.34369729, ..., 0.34369729,\n",
"           0.34369729, 0.34369729],\n",
"          [0.61522334, 0.61522334, 0.61522334, ..., 0.61522334,\n",
"           0.61522334, 0.61522334]],\n",
"\n",
"         [[0.97425583, 0.97425583, 0.97425583, ..., 0.97425583,\n",
"           0.97425583, 0.97425583],\n",
"          [0.98240233, 0.98240233, 0.98240233, ..., 0.98240233,\n",
"           0.98240233, 0.98240233],\n",
"          [0.96593192, 0.96593192, 0.96593192, ..., 0.96593192,\n",
"           0.96593192, 0.96593192],\n",
"          [0.98858014, 0.98858014, 0.98858014, ..., 0.98858014,\n",
"           0.98858014, 0.98858014]],\n",
"\n",
"         [[0.31092585, 0.31092585, 0.31092585, ..., 0.31092585,\n",
"           0.31092585, 0.31092585],\n",
"          [0.39962558, 0.39962558, 0.39962558, ..., 0.39962558,\n",
"           0.39962558, 0.39962558],\n",
"          [0.25265019, 0.25265019, 0.25265019, ..., 0.25265019,\n",
"           0.25265019, 0.25265019],\n",
"          [0.50791326, 0.50791326, 0.50791326, ..., 0.50791326,\n",
"           0.50791326, 0.50791326]],\n",
"\n",
"         ...,\n",
"\n",
"         [[0.37596846, 0.37596846, 0.37596846, ..., 0.37596846,\n",
"           0.37596846, 0.37596846],\n",
"          [0.47055246, 0.47055246, 0.47055246, ..., 0.47055246,\n",
"           0.47055246, 0.47055246],\n",
"          [0.31100422, 0.31100422, 0.31100422, ..., 0.31100422,\n",
"           0.31100422, 0.31100422],\n",
"          [0.57950811, 0.57950811, 0.57950811, ..., 0.57950811,\n",
"           0.57950811, 0.57950811]],\n",
"\n",
"         [[0.54872531, 0.54872531, 0.54872531, ..., 0.54872531,\n",
"           0.54872531, 0.54872531],\n",
"          [0.64205411, 0.64205411, 0.64205411, ..., 0.64205411,\n",
"           0.64205411, 0.64205411],\n",
"          [0.47671404, 0.47671404, 0.47671404, ..., 0.47671404,\n",
"           0.47671404, 0.47671404],\n",
"          [0.73555108, 0.73555108, 0.73555108, ..., 0.73555108,\n",
"           0.73555108, 0.73555108]],\n",
"\n",
"         [[0.82702274, 0.82702274, 0.82702274, ..., 0.82702274,\n",
"           0.82702274, 0.82702274],\n",
"          [0.87582137, 0.87582137, 0.87582137, ..., 0.87582137,\n",
"           0.87582137, 0.87582137],\n",
"          [0.78175772, 0.78175772, 0.78175772, ..., 0.78175772,\n",
"           0.78175772, 0.78175772],\n",
"          [0.9162246 , 0.9162246 , 0.9162246 , ..., 0.9162246 ,\n",
"           0.9162246 , 0.9162246 ]]],\n",
"\n",
"\n",
"        [[[0.34860267, 0.34860267, 0.34860267, ..., 0.34860267,\n",
"           0.34860267, 0.34860267],\n",
"          [0.44440894, 0.44440894, 0.44440894, ..., 0.44440894,\n",
"           0.44440894, 0.44440894],\n",
"          [0.25799757, 0.25799757, 0.25799757, ..., 0.25799757,\n",
"           0.25799757, 0.25799757],\n",
"          [0.53637973, 0.53637973, 0.53637973, ..., 0.53637973,\n",
"           0.53637973, 0.53637973]],\n",
"\n",
"         [[0.98473185, 0.98473185, 0.98473185, ..., 0.98473185,\n",
"           0.98473185, 0.98473185],\n",
"          [0.989733  , 0.989733  , 0.989733  , ..., 0.989733  ,\n",
"           0.989733  , 0.989733  ],\n",
"          [0.97669224, 0.97669224, 0.97669224, ..., 0.97669224,\n",
"           0.97669224, 0.97669224],\n",
"          [0.99287902, 0.99287902, 0.99287902, ..., 0.99287902,\n",
"           0.99287902, 0.99287902]],\n",
"\n",
"         [[0.3371221 , 0.3371221 , 0.3371221 , ..., 0.3371221 ,\n",
"           0.3371221 , 0.3371221 ],\n",
"          [0.431865  , 0.431865  , 0.431865  , ..., 0.431865  ,\n",
"           0.431865  , 0.431865  ],\n",
"          [0.2483632 , 0.2483632 , 0.2483632 , ..., 0.2483632 ,\n",
"           0.2483632 , 0.2483632 ],\n",
"          [0.52368672, 0.52368672, 0.52368672, ..., 0.52368672,\n",
"           0.52368672, 0.52368672]],\n",
"\n",
"         ...,\n",
"\n",
"         [[0.38896937, 0.38896937, 0.38896937, ..., 0.38896937,\n",
"           0.38896937, 0.38896937],\n",
"          [0.48756585, 0.48756585, 0.48756585, ..., 0.48756585,\n",
"           0.48756585, 0.48756585],\n",
"          [0.29258508, 0.29258508, 0.29258508, ..., 0.29258508,\n",
"           0.29258508, 0.29258508],\n",
"          [0.5791578 , 0.5791578 , 0.5791578 , ..., 0.5791578 ,\n",
"           0.5791578 , 0.5791578 ]],\n",
"\n",
"         [[0.54462626, 0.54462626, 0.54462626, ..., 0.54462626,\n",
"           0.54462626, 0.54462626],\n",
"          [0.64127009, 0.64127009, 0.64127009, ..., 0.64127009,\n",
"           0.64127009, 0.64127009],\n",
"          [0.43727369, 0.43727369, 0.43727369, ..., 0.43727369,\n",
"           0.43727369, 0.43727369],\n",
"          [0.72110408, 0.72110408, 0.72110408, ..., 0.72110408,\n",
"           0.72110408, 0.72110408]],\n",
"\n",
"         [[0.86547961, 0.86547961, 0.86547961, ..., 0.86547961,\n",
"           0.86547961, 0.86547961],\n",
"          [0.90580591, 0.90580591, 0.90580591, ..., 0.90580591,\n",
"           0.90580591, 0.90580591],\n",
"          [0.80695625, 0.80695625, 0.80695625, ..., 0.80695625,\n",
"           0.80695625, 0.80695625],\n",
"          [0.93292613, 0.93292613, 0.93292613, ..., 0.93292613,\n",
"           0.93292613, 0.93292613]]],\n",
"\n",
"\n",
"        [[[0.27939853, 0.27939853, 0.27939853, ..., 0.27939853,\n",
"           0.27939853, 0.27939853],\n",
"          [0.56563795, 0.56563795, 0.56563795, ..., 0.56563795,\n",
"           0.56563795, 0.56563795],\n",
"          [0.20965545, 0.20965545, 0.20965545, ..., 0.20965545,\n",
"           0.20965545, 0.20965545],\n",
"          [0.493642  , 0.493642  , 0.493642  , ..., 0.493642  ,\n",
"           0.493642  , 0.493642  ]],\n",
"\n",
"         [[0.95627397, 0.95627397, 0.95627397, ..., 0.95627397,\n",
"           0.95627397, 0.95627397],\n",
"          [0.98656842, 0.98656842, 0.98656842, ..., 0.98656842,\n",
"           0.98656842, 0.98656842],\n",
"          [0.93735302, 0.93735302, 0.93735302, ..., 0.93735302,\n",
"           0.93735302, 0.93735302],\n",
"          [0.98213902, 0.98213902, 0.98213902, ..., 0.98213902,\n",
"           0.98213902, 0.98213902]],\n",
"\n",
"         [[0.20327621, 0.20327621, 0.20327621, ..., 0.20327621,\n",
"           0.20327621, 0.20327621],\n",
"          [0.46147166, 0.46147166, 0.46147166, ..., 0.46147166,\n",
"           0.46147166, 0.46147166],\n",
"          [0.14861582, 0.14861582, 0.14861582, ..., 0.14861582,\n",
"           0.14861582, 0.14861582],\n",
"          [0.39080524, 0.39080524, 0.39080524, ..., 0.39080524,\n",
"           0.39080524, 0.39080524]],\n",
"\n",
"         ...,\n",
"\n",
"         [[0.32762106, 0.32762106, 0.32762106, ..., 0.32762106,\n",
"           0.32762106, 0.32762106],\n",
"          [0.62070895, 0.62070895, 0.62070895, ..., 0.62070895,\n",
"           0.62070895, 0.62070895],\n",
"          [0.25001713, 0.25001713, 0.25001713, ..., 0.25001713,\n",
"           0.25001713, 0.25001713],\n",
"          [0.55058863, 0.55058863, 0.55058863, ..., 0.55058863,\n",
"           0.55058863, 0.55058863]],\n",
"\n",
"         [[0.49769836, 0.49769836, 0.49769836, ..., 0.49769836,\n",
"           0.49769836, 0.49769836],\n",
"          [0.76893658, 0.76893658, 0.76893658, ..., 0.76893658,\n",
"           0.76893658, 0.76893658],\n",
"          [0.4040151 , 0.4040151 , 0.4040151 , ..., 0.4040151 ,\n",
"           0.4040151 , 0.4040151 ],\n",
"          [0.71357424, 0.71357424, 0.71357424, ..., 0.71357424,\n",
"           0.71357424, 0.71357424]],\n",
"\n",
"         [[0.90461133, 0.90461133, 0.90461133, ..., 0.90461133,\n",
"           0.90461133, 0.90461133],\n",
"          [0.96955952, 0.96955952, 0.96955952, ..., 0.96955952,\n",
"           0.96955952, 0.96955952],\n",
"          [0.86645702, 0.86645702, 0.86645702, ..., 0.86645702,\n",
"           0.86645702, 0.86645702],\n",
"          [0.95974986, 0.95974986, 0.95974986, ..., 0.95974986,\n",
"           0.95974986, 0.95974986]]]],\n",
"\n",
"\n",
"\n",
"       [[[[0.29230878, 0.29230878, 0.29230878, ..., 0.29230878,\n",
"           0.29230878, 0.29230878],\n",
"          [0.49718931, 0.49718931, 0.49718931, ..., 0.49718931,\n",
"           0.49718931, 0.49718931],\n",
"          [0.24311653, 0.24311653, 0.24311653, ..., 0.24311653,\n",
"           0.24311653, 0.24311653],\n",
"          [0.5228141 , 0.5228141 , 0.5228141 , ..., 0.5228141 ,\n",
"           0.5228141 , 0.5228141 ]],\n",
"\n",
"         [[0.9521296 , 0.9521296 , 0.9521296 , ..., 0.9521296 ,\n",
"           0.9521296 , 0.9521296 ],\n",
"          [0.97943043, 0.97943043, 0.97943043, ..., 0.97943043,\n",
"           0.97943043, 0.97943043],\n",
"          [0.93927385, 0.93927385, 0.93927385, ..., 0.93927385,\n",
"           0.93927385, 0.93927385],\n",
"          [0.98139822, 0.98139822, 0.98139822, ..., 0.98139822,\n",
"           0.98139822, 0.98139822]],\n",
"\n",
"         [[0.25408821, 0.25408821, 0.25408821, ..., 0.25408821,\n",
"           0.25408821, 0.25408821],\n",
"          [0.44918316, 0.44918316, 0.44918316, ..., 0.44918316,\n",
"           0.44918316, 0.44918316],\n",
"          [0.20942462, 0.20942462, 0.20942462, ..., 0.20942462,\n",
"           0.20942462, 0.20942462],\n",
"          [0.47466948, 0.47466948, 0.47466948, ..., 0.47466948,\n",
"           0.47466948, 0.47466948]],\n",
"\n",
"         ...,\n",
"\n",
"         [[0.24656325, 0.24656325, 0.24656325, ..., 0.24656325,\n",
"           0.24656325, 0.24656325],\n",
"          [0.43928306, 0.43928306, 0.43928306, ..., 0.43928306,\n",
"           0.43928306, 0.43928306],\n",
"          [0.20286266, 0.20286266, 0.20286266, ..., 0.20286266,\n",
"           0.20286266, 0.20286266],\n",
"          [0.46468154, 0.46468154, 0.46468154, ..., 0.46468154,\n",
"           0.46468154, 0.46468154]],\n",
"\n",
"         [[0.59373307, 0.59373307, 0.59373307, ..., 0.59373307,\n",
"           0.59373307, 0.59373307],\n",
"          [0.7777105 , 0.7777105 , 0.7777105 , ..., 0.7777105 ,\n",
"           0.7777105 , 0.7777105 ],\n",
"          [0.53194348, 0.53194348, 0.53194348, ..., 0.53194348,\n",
"           0.53194348, 0.53194348],\n",
"          [0.79493551, 0.79493551, 0.79493551, ..., 0.79493551,\n",
"           0.79493551, 0.79493551]],\n",
"\n",
"         [[0.79660123, 0.79660123, 0.79660123, ..., 0.79660123,\n",
"           0.79660123, 0.79660123],\n",
"          [0.9036226 , 0.9036226 , 0.9036226 , ..., 0.9036226 ,\n",
"           0.9036226 , 0.9036226 ],\n",
"          [0.75282097, 0.75282097, 0.75282097, ..., 0.75282097,\n",
"           0.75282097, 0.75282097],\n",
"          [0.91219239, 0.91219239, 0.91219239, ..., 0.91219239,\n",
"           0.91219239, 0.91219239]]],\n",
"\n",
"\n",
"        [[[0.36635053, 0.36635053, 0.36635053, ..., 0.36635053,\n",
"           0.36635053, 0.36635053],\n",
"          [0.50596793, 0.50596793, 0.50596793, ..., 0.50596793,\n",
"           0.50596793, 0.50596793],\n",
"          [0.30590282, 0.30590282, 0.30590282, ..., 0.30590282,\n",
"           0.30590282, 0.30590282],\n",
"          [0.5609953 , 0.5609953 , 0.5609953 , ..., 0.5609953 ,\n",
"           0.5609953 , 0.5609953 ]],\n",
"\n",
"         [[0.9588454 , 0.9588454 , 0.9588454 , ..., 0.9588454 ,\n",
"           0.9588454 , 0.9588454 ],\n",
"          [0.9763434 , 0.9763434 , 0.9763434 , ..., 0.9763434 ,\n",
"           0.9763434 , 0.9763434 ],\n",
"          [0.9466954 , 0.9466954 , 0.9466954 , ..., 0.9466954 ,\n",
"           0.9466954 , 0.9466954 ],\n",
"          [0.98095088, 0.98095088, 0.98095088, ..., 0.98095088,\n",
"           0.98095088, 0.98095088]],\n",
"\n",
"         [[0.31199312, 0.31199312, 0.31199312, ..., 0.31199312,\n",
"           0.31199312, 0.31199312],\n",
"          [0.44545801, 0.44545801, 0.44545801, ..., 0.44545801,\n",
"           0.44545801, 0.44545801],\n",
"          [0.25687839, 0.25687839, 0.25687839, ..., 0.25687839,\n",
"           0.25687839, 0.25687839],\n",
"          [0.5005725 , 0.5005725 , 0.5005725 , ..., 0.5005725 ,\n",
"           0.5005725 , 0.5005725 ]],\n",
"\n",
"         ...,\n",
"\n",
"         [[0.28453008, 0.28453008, 0.28453008, ..., 0.28453008,\n",
"           0.28453008, 0.28453008],\n",
"          [0.41330423, 0.41330423, 0.41330423, ..., 0.41330423,\n",
"           0.41330423, 0.41330423],\n",
"          [0.23262644, 0.23262644, 0.23262644, ..., 0.23262644,\n",
"           0.23262644, 0.23262644],\n",
"          [0.46779642, 0.46779642, 0.46779642, ..., 0.46779642,\n",
"           0.46779642, 0.46779642]],\n",
"\n",
"         [[0.61241248, 0.61241248, 0.61241248, ..., 0.61241248,\n",
"           0.61241248, 0.61241248],\n",
"          [0.73676902, 0.73676902, 0.73676902, ..., 0.73676902,\n",
"           0.73676902, 0.73676902],\n",
"          [0.54637264, 0.54637264, 0.54637264, ..., 0.54637264,\n",
"           0.54637264, 0.54637264],\n",
"          [0.77739888, 0.77739888, 0.77739888, ..., 0.77739888,\n",
"           0.77739888, 0.77739888]],\n",
"\n",
"         [[0.85471835, 0.85471835, 0.85471835, ..., 0.85471835,\n",
"           0.85471835, 0.85471835],\n",
"          [0.91244622, 0.91244622, 0.91244622, ..., 0.91244622,\n",
"           0.91244622, 0.91244622],\n",
"          [0.81767268, 0.81767268, 0.81767268, ..., 0.81767268,\n",
"           0.81767268, 0.81767268],\n",
"          [0.92858841, 0.92858841, 0.92858841, ..., 0.92858841,\n",
"           0.92858841, 0.92858841]]],\n",
"\n",
"\n",
"        [[[0.3837187 , 0.3837187 , 0.3837187 , ..., 0.3837187 ,\n",
"           0.3837187 , 0.3837187 ],\n",
"          [0.61631435, 0.61631435, 0.61631435, ..., 0.61631435,\n",
"           0.61631435, 0.61631435],\n",
"          [0.26100551, 0.26100551, 0.26100551, ..., 0.26100551,\n",
"           0.26100551, 0.26100551],\n",
"          [0.55675367, 0.55675367, 0.55675367, ..., 0.55675367,\n",
"           0.55675367, 0.55675367]],\n",
"\n",
"         [[0.96468772, 0.96468772, 0.96468772, ..., 0.96468772,\n",
"           0.96468772, 0.96468772],\n",
"          [0.98600969, 0.98600969, 0.98600969, ..., 0.98600969,\n",
"           0.98600969, 0.98600969],\n",
"          [0.93938132, 0.93938132, 0.93938132, ..., 0.93938132,\n",
"           0.93938132, 0.93938132],\n",
"          [0.98217845, 0.98217845, 0.98217845, ..., 0.98217845,\n",
"           0.98217845, 0.98217845]],\n",
"\n",
"         [[0.26720625, 0.26720625, 0.26720625, ..., 0.26720625,\n",
"           0.26720625, 0.26720625],\n",
"          [0.48472571, 0.48472571, 0.48472571, ..., 0.48472571,\n",
"           0.48472571, 0.48472571],\n",
"          [0.17139132, 0.17139132, 0.17139132, ..., 0.17139132,\n",
"           0.17139132, 0.17139132],\n",
"          [0.42383437, 0.42383437, 0.42383437, ..., 0.42383437,\n",
"           0.42383437, 0.42383437]],\n",
"\n",
"         ...,\n",
"\n",
"         [[0.29825557, 0.29825557, 0.29825557, ..., 0.29825557,\n",
"           0.29825557, 0.29825557],\n",
"          [0.52301093, 0.52301093, 0.52301093, ..., 0.52301093,\n",
"           0.52301093, 0.52301093],\n",
"          [0.19425843, 0.19425843, 0.19425843, ..., 0.19425843,\n",
"           0.19425843, 0.19425843],\n",
"          [0.46161878, 0.46161878, 0.46161878, ..., 0.46161878,\n",
"           0.46161878, 0.46161878]],\n",
"\n",
"         [[0.75254097, 0.75254097, 0.75254097, ..., 0.75254097,\n",
"           0.75254097, 0.75254097],\n",
"          [0.88694793, 0.88694793, 0.88694793, ..., 0.88694793,\n",
"           0.88694793, 0.88694793],\n",
"          [0.63303408, 0.63303408, 0.63303408, ..., 0.63303408,\n",
"           0.63303408, 0.63303408],\n",
"          [0.85984479, 0.85984479, 0.85984479, ..., 0.85984479,\n",
"           0.85984479, 0.85984479]],\n",
"\n",
"         [[0.85016375, 0.85016375, 0.85016375, ..., 0.85016375,\n",
"           0.85016375, 0.85016375],\n",
"          [0.93605275, 0.93605275, 0.93605275, ..., 0.93605275,\n",
"           0.93605275, 0.93605275],\n",
"          [0.7629517 , 0.7629517 , 0.7629517 , ..., 0.7629517 ,\n",
"           0.7629517 , 0.7629517 ],\n",
"          [0.91965561, 0.91965561, 0.91965561, ..., 0.91965561,\n",
"           0.91965561, 0.91965561]]],\n",
"\n",
"\n",
"        ...,\n",
"\n",
"\n",
"        [[[0.41093853, 0.41093853, 0.41093853, ..., 0.41093853,\n",
"           0.41093853, 0.41093853],\n",
"          [0.471524  , 0.471524  , 0.471524  , ..., 0.471524  ,\n",
"           0.471524  , 0.471524  ],\n",
"          [0.23719419, 0.23719419, 0.23719419, ..., 0.23719419,\n",
"           0.23719419, 0.23719419],\n",
"          [0.36942022, 0.36942022, 0.36942022, ..., 0.36942022,\n",
"           0.36942022, 0.36942022]],\n",
"\n",
"         [[0.97972067, 0.97972067, 0.97972067, ..., 0.97972067,\n",
"           0.97972067, 0.97972067],\n",
"          [0.98407364, 0.98407364, 0.98407364, ..., 0.98407364,\n",
"           0.98407364, 0.98407364],\n",
"          [0.9556224 , 0.9556224 , 0.9556224 , ..., 0.9556224 ,\n",
"           0.9556224 , 0.9556224 ],\n",
"          [0.97594463, 0.97594463, 0.97594463, ..., 0.97594463,\n",
"           0.97594463, 0.97594463]],\n",
"\n",
"         [[0.5141998 , 0.5141998 , 0.5141998 , ..., 0.5141998 ,\n",
"           0.5141998 , 0.5141998 ],\n",
"          [0.57514494, 0.57514494, 0.57514494, ..., 0.57514494,\n",
"           0.57514494, 0.57514494],\n",
"          [0.3205549 , 0.3205549 , 0.3205549 , ..., 0.3205549 ,\n",
"           0.3205549 , 0.3205549 ],\n",
"          [0.47058315, 0.47058315, 0.47058315, ..., 0.47058315,\n",
"           0.47058315, 0.47058315]],\n",
"\n",
"         ...,\n",
"\n",
"         [[0.33074744, 0.33074744, 0.33074744, ..., 0.33074744,\n",
"           0.33074744, 0.33074744],\n",
"          [0.38728313, 0.38728313, 0.38728313, ..., 0.38728313,\n",
"           0.38728313, 0.38728313],\n",
"          [0.18051772, 0.18051772, 0.18051772, ..., 0.18051772,\n",
"           0.18051772, 0.18051772],\n",
"          [0.29329707, 0.29329707, 0.29329707, ..., 0.29329707,\n",
"           0.29329707, 0.29329707]],\n",
"\n",
"         [[0.78581351, 0.78581351, 0.78581351, ..., 0.78581351,\n",
"           0.78581351, 0.78581351],\n",
"          [0.8243254 , 0.8243254 , 0.8243254 , ..., 0.8243254 ,\n",
"           0.8243254 , 0.8243254 ],\n",
"          [0.62053862, 0.62053862, 0.62053862, ..., 0.62053862,\n",
"           0.62053862, 0.62053862],\n",
"          [0.75496207, 0.75496207, 0.75496207, ..., 0.75496207,\n",
"           0.75496207, 0.75496207]],\n",
"\n",
"         [[0.8424088 , 0.8424088 , 0.8424088 , ..., 0.8424088 ,\n",
"           0.8424088 , 0.8424088 ],\n",
"          [0.87239699, 0.87239699, 0.87239699, ..., 0.87239699,\n",
"           0.87239699, 0.87239699],\n",
"          [0.70437595, 0.70437595, 0.70437595, ..., 0.70437595,\n",
"           0.70437595, 0.70437595],\n",
"          [0.81781942, 0.81781942, 0.81781942, ..., 0.81781942,\n",
"           0.81781942, 0.81781942]]],\n",
"\n",
"\n",
"        [[[0.27873963, 0.27873963, 0.27873963, ..., 0.27873963,\n",
"           0.27873963, 0.27873963],\n",
"          [0.49098128, 0.49098128, 0.49098128, ..., 0.49098128,\n",
"           0.49098128, 0.49098128],\n",
"          [0.31845388, 0.31845388, 0.31845388, ..., 0.31845388,\n",
"           0.31845388, 0.31845388],\n",
"          [0.48612488, 0.48612488, 0.48612488, ..., 0.48612488,\n",
"           0.48612488, 0.48612488]],\n",
"\n",
"         [[0.98659982, 0.98659982, 0.98659982, ..., 0.98659982,\n",
"           0.98659982, 0.98659982],\n",
"          [0.99458762, 0.99458762, 0.99458762, ..., 0.99458762,\n",
"           0.99458762, 0.99458762],\n",
"          [0.98889104, 0.98889104, 0.98889104, ..., 0.98889104,\n",
"           0.98889104, 0.98889104],\n",
"          [0.99448198, 0.99448198, 0.99448198, ..., 0.99448198,\n",
"           0.99448198, 0.99448198]],\n",
"\n",
"         [[0.2111228 , 0.2111228 , 0.2111228 , ..., 0.2111228 ,\n",
"           0.2111228 , 0.2111228 ],\n",
"          [0.40046514, 0.40046514, 0.40046514, ..., 0.40046514,\n",
"           0.40046514, 0.40046514],\n",
"          [0.24446853, 0.24446853, 0.24446853, ..., 0.24446853,\n",
"           0.24446853, 0.24446853],\n",
"          [0.39580787, 0.39580787, 0.39580787, ..., 0.39580787,\n",
"           0.39580787, 0.39580787]],\n",
"\n",
"         ...,\n",
"\n",
"         [[0.3928927 , 0.3928927 , 0.3928927 , ..., 0.3928927 ,\n",
"           0.3928927 , 0.3928927 ],\n",
"          [0.61762373, 0.61762373, 0.61762373, ..., 0.61762373,\n",
"           0.61762373, 0.61762373],\n",
"          [0.4389725 , 0.4389725 , 0.4389725 , ..., 0.4389725 ,\n",
"           0.4389725 , 0.4389725 ],\n",
"          [0.61302327, 0.61302327, 0.61302327, ..., 0.61302327,\n",
"           0.61302327, 0.61302327]],\n",
"\n",
"         [[0.44795244, 0.44795244, 0.44795244, ..., 0.44795244,\n",
"           0.44795244, 0.44795244],\n",
"          [0.66944948, 0.66944948, 0.66944948, ..., 0.66944948,\n",
"           0.66944948, 0.66944948],\n",
"          [0.49522232, 0.49522232, 0.49522232, ..., 0.49522232,\n",
"           0.49522232, 0.49522232],\n",
"          [0.66513448, 0.66513448, 0.66513448, ..., 0.66513448,\n",
"           0.66513448, 0.66513448]],\n",
"\n",
"         [[0.82057904, 0.82057904, 0.82057904, ..., 0.82057904,\n",
"           0.82057904, 0.82057904],\n",
"          [0.91945157, 0.91945157, 0.91945157, ..., 0.91945157,\n",
"           0.91945157, 0.91945157],\n",
"          [0.84685072, 0.84685072, 0.84685072, ..., 0.84685072,\n",
"           0.84685072, 0.84685072],\n",
"          [0.91800035, 0.91800035, 0.91800035, ..., 0.91800035,\n",
"           0.91800035, 0.91800035]]],\n",
"\n",
"\n",
"        [[[0.32903585, 0.32903585, 0.32903585, ..., 0.32903585,\n",
"           0.32903585, 0.32903585],\n",
"          [0.56608959, 0.56608959, 0.56608959, ..., 0.56608959,\n",
"           0.56608959, 0.56608959],\n",
"          [0.37624354, 0.37624354, 0.37624354, ..., 0.37624354,\n",
"           0.37624354, 0.37624354],\n",
"          [0.55591704, 0.55591704, 0.55591704, ..., 0.55591704,\n",
"           0.55591704, 0.55591704]],\n",
"\n",
"         [[0.96982435, 0.96982435, 0.96982435, ..., 0.96982435,\n",
"           0.96982435, 0.96982435],\n",
"          [0.98843961, 0.98843961, 0.98843961, ..., 0.98843961,\n",
"           0.98843961, 0.98843961],\n",
"          [0.97532802, 0.97532802, 0.97532802, ..., 0.97532802,\n",
"           0.97532802, 0.97532802],\n",
"          [0.98795796, 0.98795796, 0.98795796, ..., 0.98795796,\n",
"           0.98795796, 0.98795796]],\n",
"\n",
"         [[0.2498199 , 0.2498199 , 0.2498199 , ..., 0.2498199 ,\n",
"           0.2498199 , 0.2498199 ],\n",
"          [0.46975951, 0.46975951, 0.46975951, ..., 0.46975951,\n",
"           0.46975951, 0.46975951],\n",
"          [0.29058441, 0.29058441, 0.29058441, ..., 0.29058441,\n",
"           0.29058441, 0.29058441],\n",
"          [0.45948493, 0.45948493, 0.45948493, ..., 0.45948493,\n",
"           0.45948493, 0.45948493]],\n",
"\n",
"         ...,\n",
"\n",
"         [[0.30186593, 0.30186593, 0.30186593, ..., 0.30186593,\n",
"           0.30186593, 0.30186593],\n",
"          [0.53495168, 0.53495168, 0.53495168, ..., 0.53495168,\n",
"           0.53495168, 0.53495168],\n",
"          [0.34719259, 0.34719259, 0.34719259, ..., 0.34719259,\n",
"           0.34719259, 0.34719259],\n",
"          [0.5246621 , 0.5246621 , 0.5246621 , ..., 0.5246621 ,\n",
"           0.5246621 , 0.5246621 ]],\n",
"\n",
"         [[0.51113316, 0.51113316, 0.51113316, ..., 0.51113316,\n",
"           0.51113316, 0.51113316],\n",
"          [0.73555719, 0.73555719, 0.73555719, ..., 0.73555719,\n",
"           0.73555719, 0.73555719],\n",
"          [0.56256182, 0.56256182, 0.56256182, ..., 0.56256182,\n",
"           0.56256182, 0.56256182],\n",
"          [0.72744475, 0.72744475, 0.72744475, ..., 0.72744475,\n",
"           0.72744475, 0.72744475]],\n",
"\n",
"         [[0.85048785, 0.85048785, 0.85048785, ..., 0.85048785,\n",
"           0.85048785, 0.85048785],\n",
"          [0.93801629, 0.93801629, 0.93801629, ..., 0.93801629,\n",
"           0.93801629, 0.93801629],\n",
"          [0.87495055, 0.87495055, 0.87495055, ..., 0.87495055,\n",
"           0.87495055, 0.87495055],\n",
"          [0.93557076, 0.93557076, 0.93557076, ..., 0.93557076,\n",
"           0.93557076, 0.93557076]]]],\n",
"\n",
"\n",
"\n",
"       [[[[0.3611322 , 0.3611322 , 0.3611322 , ..., 0.3611322 ,\n",
"           0.3611322 , 0.3611322 ],\n",
"          [0.46407287, 0.46407287, 0.46407287, ..., 0.46407287,\n",
"           0.46407287, 0.46407287],\n",
"          [0.25932478, 0.25932478, 0.25932478, ..., 0.25932478,\n",
"           0.25932478, 0.25932478],\n",
"          [0.44781883, 0.44781883, 0.44781883, ..., 0.44781883,\n",
"           0.44781883, 0.44781883]],\n",
"\n",
"         [[0.99303658, 0.99303658, 0.99303658, ..., 0.99303658,\n",
"           0.99303658, 0.99303658],\n",
"          [0.99544332, 0.99544332, 0.99544332, ..., 0.99544332,\n",
"           0.99544332, 0.99544332],\n",
"          [0.98880544, 0.98880544, 0.98880544, ..., 0.98880544,\n",
"           0.98880544, 0.98880544],\n",
"          [0.99513622, 0.99513622, 0.99513622, ..., 0.99513622,\n",
"           0.99513622, 0.99513622]],\n",
"\n",
"         [[0.36113419, 0.36113419, 0.36113419, ..., 0.36113419,\n",
"           0.36113419, 0.36113419],\n",
"          [0.46407502, 0.46407502, 0.46407502, ..., 0.46407502,\n",
"           0.46407502, 0.46407502],\n",
"          [0.25932644, 0.25932644, 0.25932644, ..., 0.25932644,\n",
"           0.25932644, 0.25932644],\n",
"          [0.44782097, 0.44782097, 0.44782097, ..., 0.44782097,\n",
"           0.44782097, 0.44782097]],\n",
"\n",
"         ...,\n",
"\n",
"         [[0.35741229, 0.35741229, 0.35741229, ..., 0.35741229,\n",
"           0.35741229, 0.35741229],\n",
"          [0.46005618, 0.46005618, 0.46005618, ..., 0.46005618,\n",
"           0.46005618, 0.46005618],\n",
"          [0.25623296, 0.25623296, 0.25623296, ..., 0.25623296,\n",
"           0.25623296, 0.25623296],\n",
"          [0.44382632, 0.44382632, 0.44382632, ..., 0.44382632,\n",
"           0.44382632, 0.44382632]],\n",
"\n",
"         [[0.59177783, 0.59177783, 0.59177783, ..., 0.59177783,\n",
"           0.59177783, 0.59177783],\n",
"          [0.68950723, 0.68950723, 0.68950723, ..., 0.68950723,\n",
"           0.68950723, 0.68950723],\n",
"          [0.47309906, 0.47309906, 0.47309906, ..., 0.47309906,\n",
"           0.47309906, 0.47309906],\n",
"          [0.67530667, 0.67530667, 0.67530667, ..., 0.67530667,\n",
"           0.67530667, 0.67530667]],\n",
"\n",
"         [[0.81096574, 0.81096574, 0.81096574, ..., 0.81096574,\n",
"           0.81096574, 0.81096574],\n",
"          [0.86793177, 0.86793177, 0.86793177, ..., 0.86793177,\n",
"           0.86793177, 0.86793177],\n",
"          [0.72656619, 0.72656619, 0.72656619, ..., 0.72656619,\n",
"           0.72656619, 0.72656619],\n",
"          [0.86023745, 0.86023745, 0.86023745, ..., 0.86023745,\n",
"           0.86023745, 0.86023745]]],\n",
"\n",
"\n",
"        [[[0.42576834, 0.42576834, 0.42576834, ..., 0.42576834,\n",
"           0.42576834, 0.42576834],\n",
"          [0.45827876, 0.45827876, 0.45827876, ..., 0.45827876,\n",
"           0.45827876, 0.45827876],\n",
"          [0.27210715, 0.27210715, 0.27210715, ..., 0.27210715,\n",
"           0.27210715, 0.27210715],\n",
"          [0.48268575, 0.48268575, 0.48268575, ..., 0.48268575,\n",
"           0.48268575, 0.48268575]],\n",
"\n",
"         [[0.94957421, 0.94957421, 0.94957421, ..., 0.94957421,\n",
"           0.94957421, 0.94957421],\n",
"          [0.95552674, 0.95552674, 0.95552674, ..., 0.95552674,\n",
"           0.95552674, 0.95552674],\n",
"          [0.90471008, 0.90471008, 0.90471008, ..., 0.90471008,\n",
"           0.90471008, 0.90471008],\n",
"          [0.95950985, 0.95950985, 0.95950985, ..., 0.95950985,\n",
"           0.95950985, 0.95950985]],\n",
"\n",
"         [[0.32469497, 0.32469497, 0.32469497, ..., 0.32469497,\n",
"           0.32469497, 0.32469497],\n",
"          [0.35424883, 0.35424883, 0.35424883, ..., 0.35424883,\n",
"           0.35424883, 0.35424883],\n",
"          [0.19511682, 0.19511682, 0.19511682, ..., 0.19511682,\n",
"           0.19511682, 0.19511682],\n",
"          [0.37697084, 0.37697084, 0.37697084, ..., 0.37697084,\n",
"           0.37697084, 0.37697084]],\n",
"\n",
"         ...,\n",
"\n",
"         [[0.43387375, 0.43387375, 0.43387375, ..., 0.43387375,\n",
"           0.43387375, 0.43387375],\n",
"          [0.46650029, 0.46650029, 0.46650029, ..., 0.46650029,\n",
"           0.46650029, 0.46650029],\n",
"          [0.27870709, 0.27870709, 0.27870709, ..., 0.27870709,\n",
"           0.27870709, 0.27870709],\n",
"          [0.49094832, 0.49094832, 0.49094832, ..., 0.49094832,\n",
"           0.49094832, 0.49094832]],\n",
"\n",
"         [[0.68506131, 0.68506131, 0.68506131, ..., 0.68506131,\n",
"           0.68506131, 0.68506131],\n",
"          [0.71279425, 0.71279425, 0.71279425, ..., 0.71279425,\n",
"           0.71279425, 0.71279425],\n",
"          [0.52306111, 0.52306111, 0.52306111, ..., 0.52306111,\n",
"           0.52306111, 0.52306111],\n",
"          [0.73242933, 0.73242933, 0.73242933, ..., 0.73242933,\n",
"           0.73242933, 0.73242933]],\n",
"\n",
"         [[0.93390885, 0.93390885, 0.93390885, ..., 0.93390885,\n",
"           0.93390885, 0.93390885],\n",
"          [0.94159686, 0.94159686, 0.94159686, ..., 0.94159686,\n",
"           0.94159686, 0.94159686],\n",
"          [0.87691377, 0.87691377, 0.87691377, ..., 0.87691377,\n",
"           0.87691377, 0.87691377],\n",
"          [0.94675804, 0.94675804, 0.94675804, ..., 0.94675804,\n",
"           0.94675804, 0.94675804]]],\n",
"\n",
"\n",
"        [[[0.39898127, 0.39898127, 0.39898127, ..., 0.39898127,\n",
"           0.39898127, 0.39898127],\n",
"          [0.6064991 , 0.6064991 , 0.6064991 , ..., 0.6064991 ,\n",
"           0.6064991 , 0.6064991 ],\n",
"          [0.35631382, 0.35631382, 0.35631382, ..., 0.35631382,\n",
"           0.35631382, 0.35631382],\n",
"          [0.53576813, 0.53576813, 0.53576813, ..., 0.53576813,\n",
"           0.53576813, 0.53576813]],\n",
"\n",
"         [[0.96773129, 0.96773129, 0.96773129, ..., 0.96773129,\n",
"           0.96773129, 0.96773129],\n",
"          [0.9858416 , 0.9858416 , 0.9858416 , ..., 0.9858416 ,\n",
"           0.9858416 , 0.9858416 ],\n",
"          [0.96154928, 0.96154928, 0.96154928, ..., 0.96154928,\n",
"           0.96154928, 0.96154928],\n",
"          [0.98118091, 0.98118091, 0.98118091, ..., 0.98118091,\n",
"           0.98118091, 0.98118091]],\n",
"\n",
"         [[0.39923143, 0.39923143, 0.39923143, ..., 0.39923143,\n",
"           0.39923143, 0.39923143],\n",
"          [0.60674802, 0.60674802, 0.60674802, ..., 0.60674802,\n",
"           0.60674802, 0.60674802],\n",
"          [0.35655309, 0.35655309, 0.35655309, ..., 0.35655309,\n",
"           0.35655309, 0.35655309],\n",
"          [0.53602756, 0.53602756, 0.53602756, ..., 0.53602756,\n",
"           0.53602756, 0.53602756]],\n",
"\n",
"         ...,\n",
"\n",
"         [[0.27258513, 0.27258513, 0.27258513, ..., 0.27258513,\n",
"           0.27258513, 0.27258513],\n",
"          [0.46525247, 0.46525247, 0.46525247, ..., 0.46525247,\n",
"           0.46525247, 0.46525247],\n",
"          [0.23808017, 0.23808017, 0.23808017, ..., 0.23808017,\n",
"           0.23808017, 0.23808017],\n",
"          [0.39448049, 0.39448049, 0.39448049, ..., 0.39448049,\n",
"           0.39448049, 0.39448049]],\n",
"\n",
"         [[0.5240085 , 0.5240085 , 0.5240085 , ..., 0.5240085 ,\n",
"           0.5240085 , 0.5240085 ],\n",
"          [0.71878432, 0.71878432, 0.71878432, ..., 0.71878432,\n",
"           0.71878432, 0.71878432],\n",
"          [0.47861806, 0.47861806, 0.47861806, ..., 0.47861806,\n",
"           0.47861806, 0.47861806],\n",
"          [0.65681593, 0.65681593, 0.65681593, ..., 0.65681593,\n",
"           0.65681593, 0.65681593]],\n",
"\n",
"         [[0.75235642, 0.75235642, 0.75235642, ..., 0.75235642,\n",
"           0.75235642, 0.75235642],\n",
"          [0.87583334, 0.87583334, 0.87583334, ..., 0.87583334,\n",
"           0.87583334, 0.87583334],\n",
"          [0.71698028, 0.71698028, 0.71698028, ..., 0.71698028,\n",
"           0.71698028, 0.71698028],\n",
"          [0.8408075 , 0.8408075 , 0.8408075 , ..., 0.8408075 ,\n",
"           0.8408075 , 0.8408075 ]]],\n",
"\n",
"\n",
"        ...,\n",
"\n",
"\n",
"        [[[0.41490246, 0.41490246, 0.41490246, ..., 0.41490246,\n",
"           0.41490246, 0.41490246],\n",
"          [0.58450415, 0.58450415, 0.58450415, ..., 0.58450415,\n",
"           0.58450415, 0.58450415],\n",
"          [0.37053763, 0.37053763, 0.37053763, ..., 0.37053763,\n",
"           0.37053763, 0.37053763],\n",
"          [0.54497028, 0.54497028, 0.54497028, ..., 0.54497028,\n",
"           0.54497028, 0.54497028]],\n",
"\n",
"         [[0.98876773, 0.98876773, 0.98876773, ..., 0.98876773,\n",
"           0.98876773, 0.98876773],\n",
"          [0.99430636, 0.99430636, 0.99430636, ..., 0.99430636,\n",
"           0.99430636, 0.99430636],\n",
"          [0.98650025, 0.98650025, 0.98650025, ..., 0.98650025,\n",
"           0.98650025, 0.98650025],\n",
"          [0.99331892, 0.99331892, 0.99331892, ..., 0.99331892,\n",
"           0.99331892, 0.99331892]],\n",
"\n",
"         [[0.43087656, 0.43087656, 0.43087656, ..., 0.43087656,\n",
"           0.43087656, 0.43087656],\n",
"          [0.6003085 , 0.6003085 , 0.6003085 , ..., 0.6003085 ,\n",
"           0.6003085 , 0.6003085 ],\n",
"          [0.38593031, 0.38593031, 0.38593031, ..., 0.38593031,\n",
"           0.38593031, 0.38593031],\n",
"          [0.56114936, 0.56114936, 0.56114936, ..., 0.56114936,\n",
"           0.56114936, 0.56114936]],\n",
"\n",
"         ...,\n",
"\n",
"         [[0.43920209, 0.43920209, 0.43920209, ..., 0.43920209,\n",
"           0.43920209, 0.43920209],\n",
"          [0.60840805, 0.60840805, 0.60840805, ..., 0.60840805,\n",
"           0.60840805, 0.60840805],\n",
"          [0.39398858, 0.39398858, 0.39398858, ..., 0.39398858,\n",
"           0.39398858, 0.39398858],\n",
"          [0.56947333, 0.56947333, 0.56947333, ..., 0.56947333,\n",
"           0.56947333, 0.56947333]],\n",
"\n",
"         [[0.6227391 , 0.6227391 , 0.6227391 , ..., 0.6227391 ,\n",
"           0.6227391 , 0.6227391 ],\n",
"          [0.76606381, 0.76606381, 0.76606381, ..., 0.76606381,\n",
"           0.76606381, 0.76606381],\n",
"          [0.57810883, 0.57810883, 0.57810883, ..., 0.57810883,\n",
"           0.57810883, 0.57810883],\n",
"          [0.73600259, 0.73600259, 0.73600259, ..., 0.73600259,\n",
"           0.73600259, 0.73600259]],\n",
"\n",
"         [[0.81660704, 0.81660704, 0.81660704, ..., 0.81660704,\n",
"           0.81660704, 0.81660704],\n",
"          [0.89830698, 0.89830698, 0.89830698, ..., 0.89830698,\n",
"           0.89830698, 0.89830698],\n",
"          [0.78706953, 0.78706953, 0.78706953, ..., 0.78706953,\n",
"           0.78706953, 0.78706953],\n",
"          [0.88263577, 0.88263577, 0.88263577, ..., 0.88263577,\n",
"           0.88263577, 0.88263577]]],\n",
"\n",
"\n",
"        [[[0.32605625, 0.32605625, 0.32605625, ..., 0.32605625,\n",
"           0.32605625, 0.32605625],\n",
"          [0.55958368, 0.55958368, 0.55958368, ..., 0.55958368,\n",
"           0.55958368, 0.55958368],\n",
"          [0.33568331, 0.33568331, 0.33568331, ..., 0.33568331,\n",
"           0.33568331, 0.33568331],\n",
"          [0.44382289, 0.44382289, 0.44382289, ..., 0.44382289,\n",
"           0.44382289, 0.44382289]],\n",
"\n",
"         [[0.99091996, 0.99091996, 0.99091996, ..., 0.99091996,\n",
"           0.99091996, 0.99091996],\n",
"          [0.99652301, 0.99652301, 0.99652301, ..., 0.99652301,\n",
"           0.99652301, 0.99652301],\n",
"          [0.991303  , 0.991303  , 0.991303  , ..., 0.991303  ,\n",
"           0.991303  , 0.991303  ],\n",
"          [0.99447522, 0.99447522, 0.99447522, ..., 0.99447522,\n",
"           0.99447522, 0.99447522]],\n",
"\n",
"         [[0.33447933, 0.33447933, 0.33447933, ..., 0.33447933,\n",
"           0.33447933, 0.33447933],\n",
"          [0.56894664, 0.56894664, 0.56894664, ..., 0.56894664,\n",
"           0.56894664, 0.56894664],\n",
"          [0.34422806, 0.34422806, 0.34422806, ..., 0.34422806,\n",
"           0.34422806, 0.34422806],\n",
"          [0.45324226, 0.45324226, 0.45324226, ..., 0.45324226,\n",
"           0.45324226, 0.45324226]],\n",
"\n",
"         ...,\n",
"\n",
"         [[0.27791776, 0.27791776, 0.27791776, ..., 0.27791776,\n",
"           0.27791776, 0.27791776],\n",
"          [0.50268386, 0.50268386, 0.50268386, ..., 0.50268386,\n",
"           0.50268386, 0.50268386],\n",
"          [0.28672819, 0.28672819, 0.28672819, ..., 0.28672819,\n",
"           0.28672819, 0.28672819],\n",
"          [0.38831555, 0.38831555, 0.38831555, ..., 0.38831555,\n",
"           0.38831555, 0.38831555]],\n",
"\n",
"         [[0.57631247, 0.57631247, 0.57631247, ..., 0.57631247,\n",
"           0.57631247, 0.57631247],\n",
"          [0.78129058, 0.78129058, 0.78129058, ..., 0.78129058,\n",
"           0.78129058, 0.78129058],\n",
"          [0.58689393, 0.58689393, 0.58689393, ..., 0.58689393,\n",
"           0.58689393, 0.58689393],\n",
"          [0.69169794, 0.69169794, 0.69169794, ..., 0.69169794,\n",
"           0.69169794, 0.69169794]],\n",
"\n",
"         [[0.85558417, 0.85558417, 0.85558417, ..., 0.85558417,\n",
"           0.85558417, 0.85558417],\n",
"          [0.93960975, 0.93960975, 0.93960975, ..., 0.93960975,\n",
"           0.93960975, 0.93960975],\n",
"          [0.86087465, 0.86087465, 0.86087465, ..., 0.86087465,\n",
"           0.86087465, 0.86087465],\n",
"          [0.90716521, 0.90716521, 0.90716521, ..., 0.90716521,\n",
"           0.90716521, 0.90716521]]],\n",
"\n",
"\n",
"        [[[0.43919597, 0.43919597, 0.43919597, ..., 0.43919597,\n",
"           0.43919597, 0.43919597],\n",
"          [0.4573129 , 0.4573129 , 0.4573129 , ..., 0.4573129 ,\n",
"           0.4573129 , 0.4573129 ],\n",
"          [0.27618897, 0.27618897, 0.27618897, ..., 0.27618897,\n",
"           0.27618897, 0.27618897],\n",
"          [0.51587419, 0.51587419, 0.51587419, ..., 0.51587419,\n",
"           0.51587419, 0.51587419]],\n",
"\n",
"         [[0.96495737, 0.96495737, 0.96495737, ..., 0.96495737,\n",
"           0.96495737, 0.96495737],\n",
"          [0.96735202, 0.96735202, 0.96735202, ..., 0.96735202,\n",
"           0.96735202, 0.96735202],\n",
"          [0.93063595, 0.93063595, 0.93063595, ..., 0.93063595,\n",
"           0.93063595, 0.93063595],\n",
"          [0.97400374, 0.97400374, 0.97400374, ..., 0.97400374,\n",
"           0.97400374, 0.97400374]],\n",
"\n",
"         [[0.28225303, 0.28225303, 0.28225303, ..., 0.28225303,\n",
"           0.28225303, 0.28225303],\n",
"          [0.29732839, 0.29732839, 0.29732839, ..., 0.29732839,\n",
"           0.29732839, 0.29732839],\n",
"          [0.16079397, 0.16079397, 0.16079397, ..., 0.16079397,\n",
"           0.16079397, 0.16079397],\n",
"          [0.34856129, 0.34856129, 0.34856129, ..., 0.34856129,\n",
"           0.34856129, 0.34856129]],\n",
"\n",
"         ...,\n",
"\n",
"         [[0.48798296, 0.48798296, 0.48798296, ..., 0.48798296,\n",
"           0.48798296, 0.48798296],\n",
"          [0.5062955 , 0.5062955 , 0.5062955 , ..., 0.5062955 ,\n",
"           0.5062955 , 0.5062955 ],\n",
"          [0.3171075 , 0.3171075 , 0.3171075 , ..., 0.3171075 ,\n",
"           0.3171075 , 0.3171075 ],\n",
"          [0.56460348, 0.56460348, 0.56460348, ..., 0.56460348,\n",
"           0.56460348, 0.56460348]],\n",
"\n",
"         [[0.62448024, 0.62448024, 0.62448024, ..., 0.62448024,\n",
"           0.62448024, 0.62448024],\n",
"          [0.64149743, 0.64149743, 0.64149743, ..., 0.64149743,\n",
"           0.64149743, 0.64149743],\n",
"          [0.44759056, 0.44759056, 0.44759056, ..., 0.44759056,\n",
"           0.44759056, 0.44759056],\n",
"          [0.69350404, 0.69350404, 0.69350404, ..., 0.69350404,\n",
"           0.69350404, 0.69350404]],\n",
"\n",
"         [[0.88253997, 0.88253997, 0.88253997, ..., 0.88253997,\n",
"           0.88253997, 0.88253997],\n",
"          [0.88992416, 0.88992416, 0.88992416, ..., 0.88992416,\n",
"           0.88992416, 0.88992416],\n",
"          [0.7854454 , 0.7854454 , 0.7854454 , ..., 0.7854454 ,\n",
"           0.7854454 , 0.7854454 ],\n",
"          [0.91089806, 0.91089806, 0.91089806, ..., 0.91089806,\n",
"           0.91089806, 0.91089806]]]],\n",
"\n",
"\n",
"\n",
"       [[[[0.33882725, 0.33882725, 0.33882725, ..., 0.33882725,\n",
"           0.33882725, 0.33882725],\n",
"          [0.57079158, 0.57079158, 0.57079158, ..., 0.57079158,\n",
"           0.57079158, 0.57079158],\n",
"          [0.34207812, 0.34207812, 0.34207812, ..., 0.34207812,\n",
"           0.34207812, 0.34207812],\n",
"          [0.45889441, 0.45889441, 0.45889441, ..., 0.45889441,\n",
"           0.45889441, 0.45889441]],\n",
"\n",
"         [[0.9670523 , 0.9670523 , 0.9670523 , ..., 0.9670523 ,\n",
"           0.9670523 , 0.9670523 ],\n",
"          [0.98704121, 0.98704121, 0.98704121, ..., 0.98704121,\n",
"           0.98704121, 0.98704121],\n",
"          [0.96751049, 0.96751049, 0.96751049, ..., 0.96751049,\n",
"           0.96751049, 0.96751049],\n",
"          [0.97982761, 0.97982761, 0.97982761, ..., 0.97982761,\n",
"           0.97982761, 0.97982761]],\n",
"\n",
"         [[0.32168238, 0.32168238, 0.32168238, ..., 0.32168238,\n",
"           0.32168238, 0.32168238],\n",
"          [0.55170335, 0.55170335, 0.55170335, ..., 0.55170335,\n",
"           0.55170335, 0.55170335],\n",
"          [0.32484958, 0.32484958, 0.32484958, ..., 0.32484958,\n",
"           0.32484958, 0.32484958],\n",
"          [0.43971458, 0.43971458, 0.43971458, ..., 0.43971458,\n",
"           0.43971458, 0.43971458]],\n",
"\n",
"         ...,\n",
"\n",
"         [[0.37471004, 0.37471004, 0.37471004, ..., 0.37471004,\n",
"           0.37471004, 0.37471004],\n",
"          [0.60862671, 0.60862671, 0.60862671, ..., 0.60862671,\n",
"           0.60862671, 0.60862671],\n",
"          [0.37810831, 0.37810831, 0.37810831, ..., 0.37810831,\n",
"           0.37810831, 0.37810831],\n",
"          [0.49791689, 0.49791689, 0.49791689, ..., 0.49791689,\n",
"           0.49791689, 0.49791689]],\n",
"\n",
"         [[0.55778962, 0.55778962, 0.55778962, ..., 0.55778962,\n",
"           0.55778962, 0.55778962],\n",
"          [0.76598949, 0.76598949, 0.76598949, ..., 0.76598949,\n",
"           0.76598949, 0.76598949],\n",
"          [0.56135765, 0.56135765, 0.56135765, ..., 0.56135765,\n",
"           0.56135765, 0.56135765],\n",
"          [0.67610448, 0.67610448, 0.67610448, ..., 0.67610448,\n",
"           0.67610448, 0.67610448]],\n",
"\n",
"         [[0.76175935, 0.76175935, 0.76175935, ..., 0.76175935,\n",
"           0.76175935, 0.76175935],\n",
"          [0.89244434, 0.89244434, 0.89244434, ..., 0.89244434,\n",
"           0.89244434, 0.89244434],\n",
"          [0.76437682, 0.76437682, 0.76437682, ..., 0.76437682,\n",
"           0.76437682, 0.76437682],\n",
"          [0.84105246, 0.84105246, 0.84105246, ..., 0.84105246,\n",
"           0.84105246, 0.84105246]]],\n",
"\n",
"\n",
"        [[[0.46448215, 0.46448215, 0.46448215, ..., 0.46448215,\n",
"           0.46448215, 0.46448215],\n",
"          [0.58500987, 0.58500987, 0.58500987, ..., 0.58500987,\n",
"           0.58500987, 0.58500987],\n",
"          [0.29652301, 0.29652301, 0.29652301, ..., 0.29652301,\n",
"           0.29652301, 0.29652301],\n",
"          [0.55451001, 0.55451001, 0.55451001, ..., 0.55451001,\n",
"           0.55451001, 0.55451001]],\n",
"\n",
"         [[0.9892782 , 0.9892782 , 0.9892782 , ..., 0.9892782 ,\n",
"           0.9892782 , 0.9892782 ],\n",
"          [0.99337582, 0.99337582, 0.99337582, ..., 0.99337582,\n",
"           0.99337582, 0.99337582],\n",
"          [0.97818493, 0.97818493, 0.97818493, ..., 0.97818493,\n",
"           0.97818493, 0.97818493],\n",
"          [0.99250442, 0.99250442, 0.99250442, ..., 0.99250442,\n",
"           0.99250442, 0.99250442]],\n",
"\n",
"         [[0.28011381, 0.28011381, 0.28011381, ..., 0.28011381,\n",
"           0.28011381, 0.28011381],\n",
"          [0.38741005, 0.38741005, 0.38741005, ..., 0.38741005,\n",
"           0.38741005, 0.38741005],\n",
"          [0.15902555, 0.15902555, 0.15902555, ..., 0.15902555,\n",
"           0.15902555, 0.15902555],\n",
"          [0.35831708, 0.35831708, 0.35831708, ..., 0.35831708,\n",
"           0.35831708, 0.35831708]],\n",
"\n",
"         ...,\n",
"\n",
"         [[0.41229425, 0.41229425, 0.41229425, ..., 0.41229425,\n",
"           0.41229425, 0.41229425],\n",
"          [0.53275203, 0.53275203, 0.53275203, ..., 0.53275203,\n",
"           0.53275203, 0.53275203],\n",
"          [0.25424698, 0.25424698, 0.25424698, ..., 0.25424698,\n",
"           0.25424698, 0.25424698],\n",
"          [0.50168301, 0.50168301, 0.50168301, ..., 0.50168301,\n",
"           0.50168301, 0.50168301]],\n",
"\n",
"         [[0.48401999, 0.48401999, 0.48401999, ..., 0.48401999,\n",
"           0.48401999, 0.48401999],\n",
"          [0.60390032, 0.60390032, 0.60390032, ..., 0.60390032,\n",
"           0.60390032, 0.60390032],\n",
"          [0.31312689, 0.31312689, 0.31312689, ..., 0.31312689,\n",
"           0.31312689, 0.31312689],\n",
"          [0.5737773 , 0.5737773 , 0.5737773 , ..., 0.5737773 ,\n",
"           0.5737773 , 0.5737773 ]],\n",
"\n",
"         [[0.9367197 , 0.9367197 , 0.9367197 , ..., 0.9367197 ,\n",
"           0.9367197 , 0.9367197 ],\n",
"          [0.96009364, 0.96009364, 0.96009364, ..., 0.96009364,\n",
"           0.96009364, 0.96009364],\n",
"          [0.87795557, 0.87795557, 0.87795557, ..., 0.87795557,\n",
"           0.87795557, 0.87795557],\n",
"          [0.9550422 , 0.9550422 , 0.9550422 , ..., 0.9550422 ,\n",
"           0.9550422 , 0.9550422 ]]],\n",
"\n",
"\n",
"        [[[0.43217251, 0.43217251, 0.43217251, ..., 0.43217251,\n",
"           0.43217251, 0.43217251],\n",
"          [0.57116494, 0.57116494, 0.57116494, ..., 0.57116494,\n",
"           0.57116494, 0.57116494],\n",
"          [0.28112784, 0.28112784, 0.28112784, ..., 0.28112784,\n",
"           0.28112784, 0.28112784],\n",
"          [0.58953861, 0.58953861, 0.58953861, ..., 0.58953861,\n",
"           0.58953861, 0.58953861]],\n",
"\n",
"         [[0.98198767, 0.98198767, 0.98198767, ..., 0.98198767,\n",
"           0.98198767, 0.98198767],\n",
"          [0.98962699, 0.98962699, 0.98962699, ..., 0.98962699,\n",
"           0.98962699, 0.98962699],\n",
"          [0.96553177, 0.96553177, 0.96553177, ..., 0.96553177,\n",
"           0.96553177, 0.96553177],\n",
"          [0.99037361, 0.99037361, 0.99037361, ..., 0.99037361,\n",
"           0.99037361, 0.99037361]],\n",
"\n",
"         [[0.242193  , 0.242193  , 0.242193  , ..., 0.242193  ,\n",
"           0.242193  , 0.242193  ],\n",
"          [0.35868059, 0.35868059, 0.35868059, ..., 0.35868059,\n",
"           0.35868059, 0.35868059],\n",
"          [0.14105258, 0.14105258, 0.14105258, ..., 0.14105258,\n",
"           0.14105258, 0.14105258],\n",
"          [0.37621555, 0.37621555, 0.37621555, ..., 0.37621555,\n",
"           0.37621555, 0.37621555]],\n",
"\n",
"         ...,\n",
"\n",
"         [[0.31428593, 0.31428593, 0.31428593, ..., 0.31428593,\n",
"           0.31428593, 0.31428593],\n",
"          [0.44508267, 0.44508267, 0.44508267, ..., 0.44508267,\n",
"           0.44508267, 0.44508267],\n",
"          [0.19061196, 0.19061196, 0.19061196, ..., 0.19061196,\n",
"           0.19061196, 0.19061196],\n",
"          [0.46378693, 0.46378693, 0.46378693, ..., 0.46378693,\n",
"           0.46378693, 0.46378693]],\n",
"\n",
"         [[0.3850212 , 0.3850212 , 0.3850212 , ..., 0.3850212 ,\n",
"           0.3850212 , 0.3850212 ],\n",
"          [0.52281142, 0.52281142, 0.52281142, ..., 0.52281142,\n",
"           0.52281142, 0.52281142],\n",
"          [0.24339225, 0.24339225, 0.24339225, ..., 0.24339225,\n",
"           0.24339225, 0.24339225],\n",
"          [0.5415941 , 0.5415941 , 0.5415941 , ..., 0.5415941 ,\n",
"           0.5415941 , 0.5415941 ]],\n",
"\n",
"         [[0.93093208, 0.93093208, 0.93093208, ..., 0.93093208,\n",
"           0.93093208, 0.93093208],\n",
"          [0.95932805, 0.95932805, 0.95932805, ..., 0.95932805,\n",
"           0.95932805, 0.95932805],\n",
"          [0.87382549, 0.87382549, 0.87382549, ..., 0.87382549,\n",
"           0.87382549, 0.87382549],\n",
"          [0.96217212, 0.96217212, 0.96217212, ..., 0.96217212,\n",
"           0.96217212, 0.96217212]]],\n",
"\n",
"\n",
"        ...,\n",
"\n",
"\n",
"        [[[0.35989752, 0.35989752, 0.35989752, ..., 0.35989752,\n",
"           0.35989752, 0.35989752],\n",
"          [0.4694215 , 0.4694215 , 0.4694215 , ..., 0.4694215 ,\n",
"           0.4694215 , 0.4694215 ],\n",
"          [0.28227482, 0.28227482, 0.28227482, ..., 0.28227482,\n",
"           0.28227482, 0.28227482],\n",
"          [0.43124694, 0.43124694, 0.43124694, ..., 0.43124694,\n",
"           0.43124694, 0.43124694]],\n",
"\n",
"         [[0.95551066, 0.95551066, 0.95551066, ..., 0.95551066,\n",
"           0.95551066, 0.95551066],\n",
"          [0.97126095, 0.97126095, 0.97126095, ..., 0.97126095,\n",
"           0.97126095, 0.97126095],\n",
"          [0.93759073, 0.93759073, 0.93759073, ..., 0.93759073,\n",
"           0.93759073, 0.93759073],\n",
"          [0.96662617, 0.96662617, 0.96662617, ..., 0.96662617,\n",
"           0.96662617, 0.96662617]],\n",
"\n",
"         [[0.39926292, 0.39926292, 0.39926292, ..., 0.39926292,\n",
"           0.39926292, 0.39926292],\n",
"          [0.51119935, 0.51119935, 0.51119935, ..., 0.51119935,\n",
"           0.51119935, 0.51119935],\n",
"          [0.31735935, 0.31735935, 0.31735935, ..., 0.31735935,\n",
"           0.31735935, 0.31735935],\n",
"          [0.47265388, 0.47265388, 0.47265388, ..., 0.47265388,\n",
"           0.47265388, 0.47265388]],\n",
"\n",
"         ...,\n",
"\n",
"         [[0.37896278, 0.37896278, 0.37896278, ..., 0.37896278,\n",
"           0.37896278, 0.37896278],\n",
"          [0.48984868, 0.48984868, 0.48984868, ..., 0.48984868,\n",
"           0.48984868, 0.48984868],\n",
"          [0.29914982, 0.29914982, 0.29914982, ..., 0.29914982,\n",
"           0.29914982, 0.29914982],\n",
"          [0.4514263 , 0.4514263 , 0.4514263 , ..., 0.4514263 ,\n",
"           0.4514263 , 0.4514263 ]],\n",
"\n",
"         [[0.705035  , 0.705035  , 0.705035  , ..., 0.705035  ,\n",
"           0.705035  , 0.705035  ],\n",
"          [0.78996805, 0.78996805, 0.78996805, ..., 0.78996805,\n",
"           0.78996805, 0.78996805],\n",
"          [0.62574232, 0.62574232, 0.62574232, ..., 0.62574232,\n",
"           0.62574232, 0.62574232],\n",
"          [0.7632235 , 0.7632235 , 0.7632235 , ..., 0.7632235 ,\n",
"           0.7632235 , 0.7632235 ]],\n",
"\n",
"         [[0.91593476, 0.91593476, 0.91593476, ..., 0.91593476,\n",
"           0.91593476, 0.91593476],\n",
"          [0.94488775, 0.94488775, 0.94488775, ..., 0.94488775,\n",
"           0.94488775, 0.94488775],\n",
"          [0.88400907, 0.88400907, 0.88400907, ..., 0.88400907,\n",
"           0.88400907, 0.88400907],\n",
"          [0.93627877, 0.93627877, 0.93627877, ..., 0.93627877,\n",
"           0.93627877, 0.93627877]]],\n",
"\n",
"\n",
"        [[[0.24987052, 0.24987052, 0.24987052, ..., 0.24987052,\n",
"           0.24987052, 0.24987052],\n",
"          [0.40970954, 0.40970954, 0.40970954, ..., 0.40970954,\n",
"           0.40970954, 0.40970954],\n",
"          [0.22444823, 0.22444823, 0.22444823, ..., 0.22444823,\n",
"           0.22444823, 0.22444823],\n",
"          [0.38041695, 0.38041695, 0.38041695, ..., 0.38041695,\n",
"           0.38041695, 0.38041695]],\n",
"\n",
"         [[0.98702358, 0.98702358, 0.98702358, ..., 0.98702358,\n",
"           0.98702358, 0.98702358],\n",
"          [0.99373005, 0.99373005, 0.99373005, ..., 0.99373005,\n",
"           0.99373005, 0.99373005],\n",
"          [0.98509341, 0.98509341, 0.98509341, ..., 0.98509341,\n",
"           0.98509341, 0.98509341],\n",
"          [0.99291795, 0.99291795, 0.99291795, ..., 0.99291795,\n",
"           0.99291795, 0.99291795]],\n",
"\n",
"         [[0.26946694, 0.26946694, 0.26946694, ..., 0.26946694,\n",
"           0.26946694, 0.26946694],\n",
"          [0.4345792 , 0.4345792 , 0.4345792 , ..., 0.4345792 ,\n",
"           0.4345792 , 0.4345792 ],\n",
"          [0.24269594, 0.24269594, 0.24269594, ..., 0.24269594,\n",
"           0.24269594, 0.24269594],\n",
"          [0.40472766, 0.40472766, 0.40472766, ..., 0.40472766,\n",
"           0.40472766, 0.40472766]],\n",
"\n",
"         ...,\n",
"\n",
"         [[0.36534672, 0.36534672, 0.36534672, ..., 0.36534672,\n",
"           0.36534672, 0.36534672],\n",
"          [0.5453512 , 0.5453512 , 0.5453512 , ..., 0.5453512 ,\n",
"           0.5453512 , 0.5453512 ],\n",
"          [0.33339745, 0.33339745, 0.33339745, ..., 0.33339745,\n",
"           0.33339745, 0.33339745],\n",
"          [0.51481869, 0.51481869, 0.51481869, ..., 0.51481869,\n",
"           0.51481869, 0.51481869]],\n",
"\n",
"         [[0.52204904, 0.52204904, 0.52204904, ..., 0.52204904,\n",
"           0.52204904, 0.52204904],\n",
"          [0.69474347, 0.69474347, 0.69474347, ..., 0.69474347,\n",
"           0.69474347, 0.69474347],\n",
"          [0.48690967, 0.48690967, 0.48690967, ..., 0.48690967,\n",
"           0.48690967, 0.48690967],\n",
"          [0.66813845, 0.66813845, 0.66813845, ..., 0.66813845,\n",
"           0.66813845, 0.66813845]],\n",
"\n",
"         [[0.78400761, 0.78400761, 0.78400761, ..., 0.78400761,\n",
"           0.78400761, 0.78400761],\n",
"          [0.88322307, 0.88322307, 0.88322307, ..., 0.88322307,\n",
"           0.88322307, 0.88322307],\n",
"          [0.75924576, 0.75924576, 0.75924576, ..., 0.75924576,\n",
"           0.75924576, 0.75924576],\n",
"          [0.8699707 , 0.8699707 , 0.8699707 , ..., 0.8699707 ,\n",
"           0.8699707 , 0.8699707 ]]],\n",
"\n",
"\n",
"        [[[0.38820285, 0.38820285, 0.38820285, ..., 0.38820285,\n",
"           0.38820285, 0.38820285],\n",
"          [0.55787864, 0.55787864, 0.55787864, ..., 0.55787864,\n",
"           0.55787864, 0.55787864],\n",
"          [0.3158555 , 0.3158555 , 0.3158555 , ..., 0.3158555 ,\n",
"           0.3158555 , 0.3158555 ],\n",
"          [0.49478664, 0.49478664, 0.49478664, ..., 0.49478664,\n",
"           0.49478664, 0.49478664]],\n",
"\n",
"         [[0.94240611, 0.94240611, 0.94240611, ..., 0.94240611,\n",
"           0.94240611, 0.94240611],\n",
"          [0.97018426, 0.97018426, 0.97018426, ..., 0.97018426,\n",
"           0.97018426, 0.97018426],\n",
"          [0.92251421, 0.92251421, 0.92251421, ..., 0.92251421,\n",
"           0.92251421, 0.92251421],\n",
"          [0.96191253, 0.96191253, 0.96191253, ..., 0.96191253,\n",
"           0.96191253, 0.96191253]],\n",
"\n",
"         [[0.33664455, 0.33664455, 0.33664455, ..., 0.33664455,\n",
"           0.33664455, 0.33664455],\n",
"          [0.50228659, 0.50228659, 0.50228659, ..., 0.50228659,\n",
"           0.50228659, 0.50228659],\n",
"          [0.26967051, 0.26967051, 0.26967051, ..., 0.26967051,\n",
"           0.26967051, 0.26967051],\n",
"          [0.43923563, 0.43923563, 0.43923563, ..., 0.43923563,\n",
"           0.43923563, 0.43923563]],\n",
"\n",
"         ...,\n",
"\n",
"         [[0.37365979, 0.37365979, 0.37365979, ..., 0.37365979,\n",
"           0.37365979, 0.37365979],\n",
"          [0.5426168 , 0.5426168 , 0.5426168 , ..., 0.5426168 ,\n",
"           0.5426168 , 0.5426168 ],\n",
"          [0.30268184, 0.30268184, 0.30268184, ..., 0.30268184,\n",
"           0.30268184, 0.30268184],\n",
"          [0.47937935, 0.47937935, 0.47937935, ..., 0.47937935,\n",
"           0.47937935, 0.47937935]],\n",
"\n",
"         [[0.66183669, 0.66183669, 0.66183669, ..., 0.66183669,\n",
"           0.66183669, 0.66183669],\n",
"          [0.79558406, 0.79558406, 0.79558406, ..., 0.79558406,\n",
"           0.79558406, 0.79558406],\n",
"          [0.58746081, 0.58746081, 0.58746081, ..., 0.58746081,\n",
"           0.58746081, 0.58746081],\n",
"          [0.7512908 , 0.7512908 , 0.7512908 , ..., 0.7512908 ,\n",
"           0.7512908 , 0.7512908 ]],\n",
"\n",
"         [[0.91187914, 0.91187914, 0.91187914, ..., 0.91187914,\n",
"           0.91187914, 0.91187914],\n",
"          [0.95365674, 0.95365674, 0.95365674, ..., 0.95365674,\n",
"           0.95365674, 0.95365674],\n",
"          [0.88275549, 0.88275549, 0.88275549, ..., 0.88275549,\n",
"           0.88275549, 0.88275549],\n",
"          [0.94107829, 0.94107829, 0.94107829, ..., 0.94107829,\n",
"           0.94107829, 0.94107829]]]]])
• created_at :
2020-06-04T22:45:38.724507
arviz_version :
0.8.3
inference_library :
pymc3
inference_library_version :
3.8
" ], "text/plain": [ "\n", "Dimensions: (actor: 7, chain: 4, draw: 500, repetition: 18, treatment: 4)\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", " * actor (actor) int64 0 1 2 3 4 5 6\n", " * treatment (treatment) =0.8.0, we can also automatically reshape several groups at once. For instance, observed and constant data or log_likelihood could potentially benefit from the reshape too, so here we go. We start with" ] }, { "cell_type": "code", "execution_count": 23, "metadata": {}, "outputs": [ { "data": { "text/html": [ "
\n", "
xarray.Dataset
• chain: 4
• draw: 500
• experiment: 504
• chain
(chain)
int64
0 1 2 3
array([0, 1, 2, 3])
• draw
(draw)
int64
0 1 2 3 4 5 ... 495 496 497 498 499
array([  0,   1,   2, ..., 497, 498, 499])
• experiment
(experiment)
int64
0 1 2 3 4 5 ... 499 500 501 502 503
array([  0,   1,   2, ..., 501, 502, 503])
• pulled_left
(chain, draw, experiment)
float64
-0.5873 -0.8115 ... -0.1247 -0.1247
array([[[-0.58729495, -0.8115452 , -0.80748816, ..., -0.20072047,\n",
"         -0.20072047, -0.20072047],\n",
"        [-0.39601759, -1.11777939, -0.61703047, ..., -0.13521336,\n",
"         -0.13521336, -0.13521336],\n",
"        [-0.29869649, -1.35396083, -0.46467599, ..., -0.23424343,\n",
"         -0.23424343, -0.23424343],\n",
"        ...,\n",
"        [-0.53003026, -0.88815809, -0.70858513, ..., -0.24621041,\n",
"         -0.24621041, -0.24621041],\n",
"        [-0.42863549, -1.05382248, -0.58772275, ..., -0.21448582,\n",
"         -0.21448582, -0.21448582],\n",
"        [-0.32766904, -1.27511611, -0.83387688, ..., -0.14334278,\n",
"         -0.14334278, -0.14334278]],\n",
"\n",
"       [[-0.3457474 , -1.22994458, -0.68754153, ..., -0.28392784,\n",
"         -0.28392784, -0.28392784],\n",
"        [-0.45625937, -1.00416466, -0.70515484, ..., -0.20129317,\n",
"         -0.20129317, -0.20129317],\n",
"        [-0.48405177, -0.95784554, -0.95793167, ..., -0.27056056,\n",
"         -0.27056056, -0.27056056],\n",
"        ...,\n",
"        [-0.52922474, -0.88931164, -0.63775789, ..., -0.35044304,\n",
"         -0.35044304, -0.35044304],\n",
"        [-0.32675508, -1.27747718, -0.67527049, ..., -0.16623084,\n",
"         -0.16623084, -0.16623084],\n",
"        [-0.39903957, -1.11158857, -0.8349172 , ..., -0.13358791,\n",
"         -0.13358791, -0.13358791]],\n",
"\n",
"       [[-0.44805773, -1.01851119, -0.62375708, ..., -0.31942569,\n",
"         -0.31942569, -0.31942569],\n",
"        [-0.55472237, -0.85385989, -0.61300373, ..., -0.13134661,\n",
"         -0.13134661, -0.13134661],\n",
"        [-0.50912919, -0.9188408 , -0.93267193, ..., -0.33270694,\n",
"         -0.33270694, -0.33270694],\n",
"        ...,\n",
"        [-0.5359767 , -0.87971183, -0.87828266, ..., -0.23943869,\n",
"         -0.23943869, -0.23943869],\n",
"        [-0.39460863, -1.12068537, -0.82003483, ..., -0.14980638,\n",
"         -0.14980638, -0.14980638],\n",
"        [-0.57838375, -0.82280957, -0.61122236, ..., -0.24150433,\n",
"         -0.24150433, -0.24150433]],\n",
"\n",
"       [[-0.41374012, -1.0822649 , -0.84581265, ..., -0.26869439,\n",
"         -0.26869439, -0.26869439],\n",
"        [-0.62452106, -0.76683214, -0.87950054, ..., -0.13015929,\n",
"         -0.13015929, -0.13015929],\n",
"        [-0.56593762, -0.83893044, -0.8466829 , ..., -0.13487459,\n",
"         -0.13487459, -0.13487459],\n",
"        ...,\n",
"        [-0.44612699, -1.02193596, -0.63378736, ..., -0.12328795,\n",
"         -0.12328795, -0.12328795],\n",
"        [-0.28750944, -1.38681243, -0.52714056, ..., -0.27542976,\n",
"         -0.27542976, -0.27542976],\n",
"        [-0.4913545 , -0.94622727, -0.81617085, ..., -0.12470703,\n",
"         -0.12470703, -0.12470703]]])
• created_at :
2020-06-04T22:45:38.960981
arviz_version :
0.8.3
inference_library :
pymc3
inference_library_version :
3.8
" ], "text/plain": [ "\n", "Dimensions: (chain: 4, draw: 500, experiment: 504)\n", "Coordinates:\n", " * chain (chain) int64 0 1 2 3\n", " * draw (draw) int64 0 1 2 3 4 5 6 7 ... 493 494 495 496 497 498 499\n", " * experiment (experiment) int64 0 1 2 3 4 5 6 ... 498 499 500 501 502 503\n", "Data variables:\n", " pulled_left (chain, draw, experiment) float64 -0.5873 -0.8115 ... -0.1247\n", "Attributes:\n", " created_at: 2020-06-04T22:45:38.960981\n", " arviz_version: 0.8.3\n", " inference_library: pymc3\n", " inference_library_version: 3.8" ] }, "execution_count": 23, "metadata": {}, "output_type": "execute_result" } ], "source": [ "idata_11_4.log_likelihood" ] }, { "cell_type": "code", "execution_count": 24, "metadata": {}, "outputs": [], "source": [ "# these do not have actor and treatment dims so we can skip all crazy renaming\n", "def reshape_experiment_dim(ds, multi_idx):\n", " ds[\"experiment\"] = multi_idx\n", " return ds.unstack(\"experiment\")" ] }, { "cell_type": "code", "execution_count": 25, "metadata": {}, "outputs": [], "source": [ "multi_idx = pd.MultiIndex.from_frame(\n", " d[[\"actor\", \"treatment\", \"repetition\"]] # no renamig here either\n", ")\n", "idata_11_4 = idata_11_4.map(\n", " reshape_experiment_dim, \n", " groups=(\"constant_data\", \"observed_data\", \"log_likelihood\"), \n", " args=(multi_idx,)\n", ")" ] }, { "cell_type": "code", "execution_count": 26, "metadata": {}, "outputs": [ { "data": { "text/html": [ "
\n", "
xarray.Dataset
• actor: 7
• chain: 4
• draw: 500
• repetition: 18
• treatment: 4
• chain
(chain)
int64
0 1 2 3
array([0, 1, 2, 3])
• draw
(draw)
int64
0 1 2 3 4 5 ... 495 496 497 498 499
array([  0,   1,   2, ..., 497, 498, 499])
• actor
(actor)
int64
0 1 2 3 4 5 6
array([0, 1, 2, 3, 4, 5, 6])
• treatment
(treatment)
int64
0 1 2 3
array([0, 1, 2, 3])
• repetition
(repetition)
int64
0 1 2 3 4 5 6 ... 12 13 14 15 16 17
array([ 0,  1,  2,  3,  4,  5,  6,  7,  8,  9, 10, 11, 12, 13, 14, 15, 16, 17])
• pulled_left
(chain, draw, actor, treatment, repetition)
float64
-0.5873 -0.8115 ... -0.06073
array([[[[[-5.87294954e-01, -8.11545196e-01, -5.87294954e-01, ...,\n",
"           -5.87294954e-01, -5.87294954e-01, -5.87294954e-01],\n",
"          [-8.07488156e-01, -5.90548869e-01, -5.90548869e-01, ...,\n",
"           -5.90548869e-01, -5.90548869e-01, -5.90548869e-01],\n",
"          [-1.15230173e+00, -3.79664032e-01, -3.79664032e-01, ...,\n",
"           -3.79664032e-01, -3.79664032e-01, -3.79664032e-01],\n",
"          [-6.05433084e-01, -6.05433084e-01, -6.05433084e-01, ...,\n",
"           -6.05433084e-01, -6.05433084e-01, -7.89301276e-01]],\n",
"\n",
"         [[-4.49259596e-02, -4.49259596e-02, -4.49259596e-02, ...,\n",
"           -4.49259596e-02, -4.49259596e-02, -4.49259596e-02],\n",
"          [-2.91301267e-02, -2.91301267e-02, -2.91301267e-02, ...,\n",
"           -2.91301267e-02, -2.91301267e-02, -2.91301267e-02],\n",
"          [-7.65122078e-02, -7.65122078e-02, -7.65122078e-02, ...,\n",
"           -7.65122078e-02, -7.65122078e-02, -7.65122078e-02],\n",
"          [-4.31857355e-02, -4.31857355e-02, -4.31857355e-02, ...,\n",
"           -4.31857355e-02, -4.31857355e-02, -4.31857355e-02]],\n",
"\n",
"         [[-4.78369053e-01, -4.78369053e-01, -9.67040537e-01, ...,\n",
"           -4.78369053e-01, -9.67040537e-01, -9.67040537e-01],\n",
"          [-6.69687994e-01, -6.69687994e-01, -6.69687994e-01, ...,\n",
"           -7.17169948e-01, -7.17169948e-01, -7.17169948e-01],\n",
"          [-3.03429241e-01, -3.03429241e-01, -1.34048818e+00, ...,\n",
"           -3.03429241e-01, -1.34048818e+00, -3.03429241e-01],\n",
"          [-4.93915321e-01, -4.93915321e-01, -4.93915321e-01, ...,\n",
"           -9.42204755e-01, -4.93915321e-01, -9.42204755e-01]],\n",
"\n",
"         ...,\n",
"\n",
"         [[-5.45081494e-01, -8.67011478e-01, -5.45081494e-01, ...,\n",
"           -5.45081494e-01, -8.67011478e-01, -5.45081494e-01],\n",
"          [-6.35294210e-01, -6.35294210e-01, -6.35294210e-01, ...,\n",
"           -7.54553756e-01, -7.54553756e-01, -6.35294210e-01],\n",
"          [-1.22014198e+00, -3.49824539e-01, -3.49824539e-01, ...,\n",
"           -1.22014198e+00, -3.49824539e-01, -3.49824539e-01],\n",
"          [-5.62249313e-01, -5.62249313e-01, -5.62249313e-01, ...,\n",
"           -8.43797247e-01, -8.43797247e-01, -8.43797247e-01]],\n",
"\n",
"         [[-1.15266520e+00, -1.15266520e+00, -3.79496228e-01, ...,\n",
"           -3.79496228e-01, -1.15266520e+00, -3.79496228e-01],\n",
"          [-2.59977122e-01, -2.59977122e-01, -1.47433562e+00, ...,\n",
"           -2.59977122e-01, -2.59977122e-01, -2.59977122e-01],\n",
"          [-8.11840527e-01, -8.11840527e-01, -8.11840527e-01, ...,\n",
"           -5.87059012e-01, -5.87059012e-01, -8.11840527e-01],\n",
"          [-3.66919111e-01, -3.66919111e-01, -1.18047013e+00, ...,\n",
"           -1.18047013e+00, -3.66919111e-01, -3.66919111e-01]],\n",
"\n",
"         [[-1.20847792e-01, -2.17303891e+00, -1.20847792e-01, ...,\n",
"           -1.20847792e-01, -2.17303891e+00, -1.20847792e-01],\n",
"          [-7.93934051e-02, -7.93934051e-02, -2.57277405e+00, ...,\n",
"           -7.93934051e-02, -7.93934051e-02, -7.93934051e-02],\n",
"          [-2.00720471e-01, -2.00720471e-01, -2.00720471e-01, ...,\n",
"           -2.00720471e-01, -2.00720471e-01, -2.00720471e-01],\n",
"          [-1.16332453e-01, -1.16332453e-01, -1.16332453e-01, ...,\n",
"           -1.16332453e-01, -1.16332453e-01, -1.16332453e-01]]],\n",
"\n",
"\n",
"        [[[-3.96017588e-01, -1.11777939e+00, -3.96017588e-01, ...,\n",
"           -3.96017588e-01, -3.96017588e-01, -3.96017588e-01],\n",
"          [-6.17030472e-01, -7.75538526e-01, -7.75538526e-01, ...,\n",
"           -7.75538526e-01, -7.75538526e-01, -7.75538526e-01],\n",
"          [-1.31378262e+00, -3.13070115e-01, -3.13070115e-01, ...,\n",
"           -3.13070115e-01, -3.13070115e-01, -3.13070115e-01],\n",
"          [-7.03085022e-01, -7.03085022e-01, -7.03085022e-01, ...,\n",
"           -7.03085022e-01, -7.03085022e-01, -6.83307129e-01]],\n",
"\n",
"         [[-4.69302641e-03, -4.69302641e-03, -4.69302641e-03, ...,\n",
"           -4.69302641e-03, -4.69302641e-03, -4.69302641e-03],\n",
"          [-2.67469059e-03, -2.67469059e-03, -2.67469059e-03, ...,\n",
"           -2.67469059e-03, -2.67469059e-03, -2.67469059e-03],\n",
"          [-6.19829993e-03, -6.19829993e-03, -6.19829993e-03, ...,\n",
"           -6.19829993e-03, -6.19829993e-03, -6.19829993e-03],\n",
"          [-2.23841044e-03, -2.23841044e-03, -2.23841044e-03, ...,\n",
"           -2.23841044e-03, -2.23841044e-03, -2.23841044e-03]],\n",
"\n",
"         [[-3.01601949e-01, -3.01601949e-01, -1.34566087e+00, ...,\n",
"           -3.01601949e-01, -1.34566087e+00, -1.34566087e+00],\n",
"          [-4.81367172e-01, -4.81367172e-01, -4.81367172e-01, ...,\n",
"           -9.62172345e-01, -9.62172345e-01, -9.62172345e-01],\n",
"          [-2.36124954e-01, -2.36124954e-01, -1.55913458e+00, ...,\n",
"           -2.36124954e-01, -1.55913458e+00, -2.36124954e-01],\n",
"          [-5.53283945e-01, -5.53283945e-01, -5.53283945e-01, ...,\n",
"           -8.55803170e-01, -5.53283945e-01, -8.55803170e-01]],\n",
"\n",
"         ...,\n",
"\n",
"         [[-4.11047618e-01, -1.08753990e+00, -4.11047618e-01, ...,\n",
"           -4.11047618e-01, -1.08753990e+00, -4.11047618e-01],\n",
"          [-7.51368464e-01, -7.51368464e-01, -7.51368464e-01, ...,\n",
"           -6.38129927e-01, -6.38129927e-01, -7.51368464e-01],\n",
"          [-1.28088440e+00, -3.25441417e-01, -3.25441417e-01, ...,\n",
"           -1.28088440e+00, -3.25441417e-01, -3.25441417e-01],\n",
"          [-7.26199688e-01, -7.26199688e-01, -7.26199688e-01, ...,\n",
"           -6.61152278e-01, -6.61152278e-01, -6.61152278e-01]],\n",
"\n",
"         [[-8.73035366e-01, -8.73035366e-01, -5.40738256e-01, ...,\n",
"           -5.40738256e-01, -8.73035366e-01, -5.40738256e-01],\n",
"          [-3.42441937e-01, -3.42441937e-01, -1.23799279e+00, ...,\n",
"           -3.42441937e-01, -3.42441937e-01, -3.42441937e-01],\n",
"          [-7.20176071e-01, -7.20176071e-01, -7.20176071e-01, ...,\n",
"           -6.66829666e-01, -6.66829666e-01, -7.20176071e-01],\n",
"          [-2.93933727e-01, -2.93933727e-01, -1.36777053e+00, ...,\n",
"           -1.36777053e+00, -2.93933727e-01, -2.93933727e-01]],\n",
"\n",
"         [[-1.03943721e-01, -2.31542739e+00, -1.03943721e-01, ...,\n",
"           -1.03943721e-01, -2.31542739e+00, -1.03943721e-01],\n",
"          [-6.04983558e-02, -6.04983558e-02, -2.83523577e+00, ...,\n",
"           -6.04983558e-02, -6.04983558e-02, -6.04983558e-02],\n",
"          [-1.35213360e-01, -1.35213360e-01, -1.35213360e-01, ...,\n",
"           -1.35213360e-01, -1.35213360e-01, -1.35213360e-01],\n",
"          [-5.08658209e-02, -5.08658209e-02, -5.08658209e-02, ...,\n",
"           -5.08658209e-02, -5.08658209e-02, -5.08658209e-02]]],\n",
"\n",
"\n",
"        [[[-2.98696490e-01, -1.35396083e+00, -2.98696490e-01, ...,\n",
"           -2.98696490e-01, -2.98696490e-01, -2.98696490e-01],\n",
"          [-4.64675993e-01, -9.89772211e-01, -9.89772211e-01, ...,\n",
"           -9.89772211e-01, -9.89772211e-01, -9.89772211e-01],\n",
"          [-1.37360473e+00, -2.91947989e-01, -2.91947989e-01, ...,\n",
"           -2.91947989e-01, -2.91947989e-01, -2.91947989e-01],\n",
"          [-4.89204521e-01, -4.89204521e-01, -4.89204521e-01, ...,\n",
"           -4.89204521e-01, -4.89204521e-01, -9.49624995e-01]],\n",
"\n",
"         [[-1.02869391e-01, -1.02869391e-01, -1.02869391e-01, ...,\n",
"           -1.02869391e-01, -1.02869391e-01, -1.02869391e-01],\n",
"          [-6.18122927e-02, -6.18122927e-02, -6.18122927e-02, ...,\n",
"           -6.18122927e-02, -6.18122927e-02, -6.18122927e-02],\n",
"          [-1.05480321e-01, -1.05480321e-01, -1.05480321e-01, ...,\n",
"           -1.05480321e-01, -1.05480321e-01, -1.05480321e-01],\n",
"          [-5.80512260e-02, -5.80512260e-02, -5.80512260e-02, ...,\n",
"           -5.80512260e-02, -5.80512260e-02, -5.80512260e-02]],\n",
"\n",
"         [[-4.50665113e-01, -4.50665113e-01, -1.01391513e+00, ...,\n",
"           -4.50665113e-01, -1.01391513e+00, -1.01391513e+00],\n",
"          [-6.76743026e-01, -6.76743026e-01, -6.76743026e-01, ...,\n",
"           -7.09824926e-01, -7.09824926e-01, -7.09824926e-01],\n",
"          [-4.41170379e-01, -4.41170379e-01, -1.03081280e+00, ...,\n",
"           -4.41170379e-01, -1.03081280e+00, -4.41170379e-01],\n",
"          [-7.09068869e-01, -7.09068869e-01, -7.09068869e-01, ...,\n",
"           -6.77475025e-01, -7.09068869e-01, -6.77475025e-01]],\n",
"\n",
"         ...,\n",
"\n",
"         [[-4.62207992e-01, -9.93958570e-01, -4.62207992e-01, ...,\n",
"           -4.62207992e-01, -9.93958570e-01, -4.62207992e-01],\n",
"          [-6.93938723e-01, -6.93938723e-01, -6.93938723e-01, ...,\n",
"           -6.92356264e-01, -6.92356264e-01, -6.93938723e-01],\n",
"          [-1.01066392e+00, -4.52520942e-01, -4.52520942e-01, ...,\n",
"           -1.01066392e+00, -4.52520942e-01, -4.52520942e-01],\n",
"          [-7.25191336e-01, -7.25191336e-01, -7.25191336e-01, ...,\n",
"           -6.62098051e-01, -6.62098051e-01, -6.62098051e-01]],\n",
"\n",
"         [[-8.62305228e-01, -8.62305228e-01, -5.48506239e-01, ...,\n",
"           -5.48506239e-01, -8.62305228e-01, -5.48506239e-01],\n",
"          [-3.57675335e-01, -3.57675335e-01, -1.20164244e+00, ...,\n",
"           -3.57675335e-01, -3.57675335e-01, -3.57675335e-01],\n",
"          [-8.47140450e-01, -8.47140450e-01, -8.47140450e-01, ...,\n",
"           -5.59733862e-01, -5.59733862e-01, -8.47140450e-01],\n",
"          [-3.38663320e-01, -3.38663320e-01, -1.24730617e+00, ...,\n",
"           -1.24730617e+00, -3.38663320e-01, -3.38663320e-01]],\n",
"\n",
"         [[-2.28789127e-01, -1.58716904e+00, -2.28789127e-01, ...,\n",
"           -2.28789127e-01, -1.58716904e+00, -2.28789127e-01],\n",
"          [-1.40884200e-01, -1.40884200e-01, -2.02943223e+00, ...,\n",
"           -1.40884200e-01, -1.40884200e-01, -1.40884200e-01],\n",
"          [-2.34243425e-01, -2.34243425e-01, -2.34243425e-01, ...,\n",
"           -2.34243425e-01, -2.34243425e-01, -2.34243425e-01],\n",
"          [-1.32620110e-01, -1.32620110e-01, -1.32620110e-01, ...,\n",
"           -1.32620110e-01, -1.32620110e-01, -1.32620110e-01]]],\n",
"\n",
"\n",
"        ...,\n",
"\n",
"\n",
"        [[[-5.30030262e-01, -8.88158085e-01, -5.30030262e-01, ...,\n",
"           -5.30030262e-01, -5.30030262e-01, -5.30030262e-01],\n",
"          [-7.08585130e-01, -6.77943942e-01, -6.77943942e-01, ...,\n",
"           -6.77943942e-01, -6.77943942e-01, -6.77943942e-01],\n",
"          [-1.06799399e+00, -4.21133142e-01, -4.21133142e-01, ...,\n",
"           -4.21133142e-01, -4.21133142e-01, -4.21133142e-01],\n",
"          [-9.55092216e-01, -9.55092216e-01, -9.55092216e-01, ...,\n",
"           -9.55092216e-01, -9.55092216e-01, -4.85769923e-01]],\n",
"\n",
"         [[-2.60813555e-02, -2.60813555e-02, -2.60813555e-02, ...,\n",
"           -2.60813555e-02, -2.60813555e-02, -2.60813555e-02],\n",
"          [-1.77543474e-02, -1.77543474e-02, -1.77543474e-02, ...,\n",
"           -1.77543474e-02, -1.77543474e-02, -1.77543474e-02],\n",
"          [-3.46619220e-02, -3.46619220e-02, -3.46619220e-02, ...,\n",
"           -3.46619220e-02, -3.46619220e-02, -3.46619220e-02],\n",
"          [-1.14855685e-02, -1.14855685e-02, -1.14855685e-02, ...,\n",
"           -1.14855685e-02, -1.14855685e-02, -1.14855685e-02]],\n",
"\n",
"         [[-3.72406390e-01, -3.72406390e-01, -1.16820083e+00, ...,\n",
"           -3.72406390e-01, -1.16820083e+00, -1.16820083e+00],\n",
"          [-5.10201788e-01, -5.10201788e-01, -5.10201788e-01, ...,\n",
"           -9.17227216e-01, -9.17227216e-01, -9.17227216e-01],\n",
"          [-2.91221920e-01, -2.91221920e-01, -1.37574939e+00, ...,\n",
"           -2.91221920e-01, -1.37574939e+00, -2.91221920e-01],\n",
"          [-7.09100274e-01, -7.09100274e-01, -7.09100274e-01, ...,\n",
"           -6.77444597e-01, -7.09100274e-01, -6.77444597e-01]],\n",
"\n",
"         ...,\n",
"\n",
"         [[-4.71554374e-01, -9.78250010e-01, -4.71554374e-01, ...,\n",
"           -4.71554374e-01, -9.78250010e-01, -4.71554374e-01],\n",
"          [-7.53847823e-01, -7.53847823e-01, -7.53847823e-01, ...,\n",
"           -6.35921198e-01, -6.35921198e-01, -7.53847823e-01],\n",
"          [-1.16794880e+00, -3.72520133e-01, -3.72520133e-01, ...,\n",
"           -1.16794880e+00, -3.72520133e-01, -3.72520133e-01],\n",
"          [-8.66330095e-01, -8.66330095e-01, -8.66330095e-01, ...,\n",
"           -5.45575615e-01, -5.45575615e-01, -5.45575615e-01]],\n",
"\n",
"         [[-7.95679052e-01, -7.95679052e-01, -6.00157312e-01, ...,\n",
"           -6.00157312e-01, -7.95679052e-01, -6.00157312e-01],\n",
"          [-4.43082697e-01, -4.43082697e-01, -1.02737345e+00, ...,\n",
"           -4.43082697e-01, -4.43082697e-01, -4.43082697e-01],\n",
"          [-6.47627187e-01, -6.47627187e-01, -6.47627187e-01, ...,\n",
"           -7.40838474e-01, -7.40838474e-01, -6.47627187e-01],\n",
"          [-3.07135297e-01, -3.07135297e-01, -1.33010715e+00, ...,\n",
"           -1.33010715e+00, -3.07135297e-01, -3.07135297e-01]],\n",
"\n",
"         [[-1.89923084e-01, -1.75459516e+00, -1.89923084e-01, ...,\n",
"           -1.89923084e-01, -1.75459516e+00, -1.89923084e-01],\n",
"          [-1.32593122e-01, -1.32593122e-01, -2.08603420e+00, ...,\n",
"           -1.32593122e-01, -1.32593122e-01, -1.32593122e-01],\n",
"          [-2.46210409e-01, -2.46210409e-01, -2.46210409e-01, ...,\n",
"           -2.46210409e-01, -2.46210409e-01, -2.46210409e-01],\n",
"          [-8.74937425e-02, -8.74937425e-02, -8.74937425e-02, ...,\n",
"           -8.74937425e-02, -8.74937425e-02, -8.74937425e-02]]],\n",
"\n",
"\n",
"        [[[-4.28635487e-01, -1.05382248e+00, -4.28635487e-01, ...,\n",
"           -4.28635487e-01, -4.28635487e-01, -4.28635487e-01],\n",
"          [-5.87722753e-01, -8.11010112e-01, -8.11010112e-01, ...,\n",
"           -8.11010112e-01, -8.11010112e-01, -8.11010112e-01],\n",
"          [-1.35480513e+00, -2.98402755e-01, -2.98402755e-01, ...,\n",
"           -2.98402755e-01, -2.98402755e-01, -2.98402755e-01],\n",
"          [-7.68689437e-01, -7.68689437e-01, -7.68689437e-01, ...,\n",
"           -7.68689437e-01, -7.68689437e-01, -6.22912924e-01]],\n",
"\n",
"         [[-1.53859060e-02, -1.53859060e-02, -1.53859060e-02, ...,\n",
"           -1.53859060e-02, -1.53859060e-02, -1.53859060e-02],\n",
"          [-1.03200706e-02, -1.03200706e-02, -1.03200706e-02, ...,\n",
"           -1.03200706e-02, -1.03200706e-02, -1.03200706e-02],\n",
"          [-2.35836831e-02, -2.35836831e-02, -2.35836831e-02, ...,\n",
"           -2.35836831e-02, -2.35836831e-02, -2.35836831e-02],\n",
"          [-7.14645087e-03, -7.14645087e-03, -7.14645087e-03, ...,\n",
"           -7.14645087e-03, -7.14645087e-03, -7.14645087e-03]],\n",
"\n",
"         [[-4.11164464e-01, -4.11164464e-01, -1.08731011e+00, ...,\n",
"           -4.11164464e-01, -1.08731011e+00, -1.08731011e+00],\n",
"          [-5.65396220e-01, -5.65396220e-01, -5.65396220e-01, ...,\n",
"           -8.39642229e-01, -8.39642229e-01, -8.39642229e-01],\n",
"          [-2.85502054e-01, -2.85502054e-01, -1.39286308e+00, ...,\n",
"           -2.85502054e-01, -1.39286308e+00, -2.85502054e-01],\n",
"          [-7.41679495e-01, -7.41679495e-01, -7.41679495e-01, ...,\n",
"           -6.46861631e-01, -7.41679495e-01, -6.46861631e-01]],\n",
"\n",
"         ...,\n",
"\n",
"         [[-4.92608194e-01, -9.44254673e-01, -4.92608194e-01, ...,\n",
"           -4.92608194e-01, -9.44254673e-01, -4.92608194e-01],\n",
"          [-7.18329914e-01, -7.18329914e-01, -7.18329914e-01, ...,\n",
"           -6.68583070e-01, -6.68583070e-01, -7.18329914e-01],\n",
"          [-1.22899977e+00, -3.46137914e-01, -3.46137914e-01, ...,\n",
"           -1.22899977e+00, -3.46137914e-01, -3.46137914e-01],\n",
"          [-8.65497332e-01, -8.65497332e-01, -8.65497332e-01, ...,\n",
"           -5.46180303e-01, -5.46180303e-01, -5.46180303e-01]],\n",
"\n",
"         [[-7.86636789e-01, -7.86636789e-01, -6.07655482e-01, ...,\n",
"           -6.07655482e-01, -7.86636789e-01, -6.07655482e-01],\n",
"          [-4.44304560e-01, -4.44304560e-01, -1.02518550e+00, ...,\n",
"           -4.44304560e-01, -4.44304560e-01, -4.44304560e-01],\n",
"          [-5.74961904e-01, -5.74961904e-01, -5.74961904e-01, ...,\n",
"           -8.27195978e-01, -8.27195978e-01, -5.74961904e-01],\n",
"          [-3.26971797e-01, -3.26971797e-01, -1.27691661e+00, ...,\n",
"           -1.27691661e+00, -3.26971797e-01, -3.26971797e-01]],\n",
"\n",
"         [[-1.44471467e-01, -2.00603947e+00, -1.44471467e-01, ...,\n",
"           -1.44471467e-01, -2.00603947e+00, -1.44471467e-01],\n",
"          [-9.89302214e-02, -9.89302214e-02, -2.36239786e+00, ...,\n",
"           -9.89302214e-02, -9.89302214e-02, -9.89302214e-02],\n",
"          [-2.14485823e-01, -2.14485823e-01, -2.14485823e-01, ...,\n",
"           -2.14485823e-01, -2.14485823e-01, -2.14485823e-01],\n",
"          [-6.94292537e-02, -6.94292537e-02, -6.94292537e-02, ...,\n",
"           -6.94292537e-02, -6.94292537e-02, -6.94292537e-02]]],\n",
"\n",
"\n",
"        [[[-3.27669036e-01, -1.27511611e+00, -3.27669036e-01, ...,\n",
"           -3.27669036e-01, -3.27669036e-01, -3.27669036e-01],\n",
"          [-8.33876883e-01, -5.69801065e-01, -5.69801065e-01, ...,\n",
"           -5.69801065e-01, -5.69801065e-01, -5.69801065e-01],\n",
"          [-1.56228983e+00, -2.35286285e-01, -2.35286285e-01, ...,\n",
"           -2.35286285e-01, -2.35286285e-01, -2.35286285e-01],\n",
"          [-6.80511355e-01, -6.80511355e-01, -6.80511355e-01, ...,\n",
"           -6.80511355e-01, -6.80511355e-01, -7.05944716e-01]],\n",
"\n",
"         [[-4.47108253e-02, -4.47108253e-02, -4.47108253e-02, ...,\n",
"           -4.47108253e-02, -4.47108253e-02, -4.47108253e-02],\n",
"          [-1.35226025e-02, -1.35226025e-02, -1.35226025e-02, ...,\n",
"           -1.35226025e-02, -1.35226025e-02, -1.35226025e-02],\n",
"          [-6.46953144e-02, -6.46953144e-02, -6.46953144e-02, ...,\n",
"           -6.46953144e-02, -6.46953144e-02, -6.46953144e-02],\n",
"          [-1.80224086e-02, -1.80224086e-02, -1.80224086e-02, ...,\n",
"           -1.80224086e-02, -1.80224086e-02, -1.80224086e-02]],\n",
"\n",
"         [[-2.27247217e-01, -2.27247217e-01, -1.59318960e+00, ...,\n",
"           -2.27247217e-01, -1.59318960e+00, -1.59318960e+00],\n",
"          [-6.18915147e-01, -6.18915147e-01, -6.18915147e-01, ...,\n",
"           -7.73334646e-01, -7.73334646e-01, -7.73334646e-01],\n",
"          [-1.60891811e-01, -1.60891811e-01, -1.90639067e+00, ...,\n",
"           -1.60891811e-01, -1.90639067e+00, -1.60891811e-01],\n",
"          [-4.95617265e-01, -4.95617265e-01, -4.95617265e-01, ...,\n",
"           -9.39545942e-01, -4.95617265e-01, -9.39545942e-01]],\n",
"\n",
"         ...,\n",
"\n",
"         [[-3.96933200e-01, -1.11589764e+00, -3.96933200e-01, ...,\n",
"           -3.96933200e-01, -1.11589764e+00, -3.96933200e-01],\n",
"          [-4.76892983e-01, -4.76892983e-01, -4.76892983e-01, ...,\n",
"           -9.69451432e-01, -9.69451432e-01, -4.76892983e-01],\n",
"          [-1.38622583e+00, -2.87704918e-01, -2.87704918e-01, ...,\n",
"           -1.38622583e+00, -2.87704918e-01, -2.87704918e-01],\n",
"          [-7.99816612e-01, -7.99816612e-01, -7.99816612e-01, ...,\n",
"           -5.96767342e-01, -5.96767342e-01, -5.96767342e-01]],\n",
"\n",
"         [[-6.88554471e-01, -6.88554471e-01, -6.97761080e-01, ...,\n",
"           -6.97761080e-01, -6.88554471e-01, -6.97761080e-01],\n",
"          [-2.62746783e-01, -2.62746783e-01, -1.46506306e+00, ...,\n",
"           -2.62746783e-01, -2.62746783e-01, -2.62746783e-01],\n",
"          [-5.17539947e-01, -5.17539947e-01, -5.17539947e-01, ...,\n",
"           -9.06303027e-01, -9.06303027e-01, -5.17539947e-01],\n",
"          [-3.37468799e-01, -3.37468799e-01, -1.25027590e+00, ...,\n",
"           -1.25027590e+00, -3.37468799e-01, -3.37468799e-01]],\n",
"\n",
"         [[-1.00249896e-01, -2.34979548e+00, -1.00249896e-01, ...,\n",
"           -1.00249896e-01, -2.34979548e+00, -1.00249896e-01],\n",
"          [-3.09134165e-02, -3.09134165e-02, -3.49198189e+00, ...,\n",
"           -3.09134165e-02, -3.09134165e-02, -3.09134165e-02],\n",
"          [-1.43342775e-01, -1.43342775e-01, -1.43342775e-01, ...,\n",
"           -1.43342775e-01, -1.43342775e-01, -1.43342775e-01],\n",
"          [-4.10825874e-02, -4.10825874e-02, -4.10825874e-02, ...,\n",
"           -4.10825874e-02, -4.10825874e-02, -4.10825874e-02]]]],\n",
"\n",
"\n",
"\n",
"       [[[[-3.45747405e-01, -1.22994458e+00, -3.45747405e-01, ...,\n",
"           -3.45747405e-01, -3.45747405e-01, -3.45747405e-01],\n",
"          [-6.87541535e-01, -6.98784427e-01, -6.98784427e-01, ...,\n",
"           -6.98784427e-01, -6.98784427e-01, -6.98784427e-01],\n",
"          [-1.41421441e+00, -2.78545971e-01, -2.78545971e-01, ...,\n",
"           -2.78545971e-01, -2.78545971e-01, -2.78545971e-01],\n",
"          [-7.39849142e-01, -7.39849142e-01, -7.39849142e-01, ...,\n",
"           -7.39849142e-01, -7.39849142e-01, -6.48529322e-01]],\n",
"\n",
"         [[-4.90541153e-02, -4.90541153e-02, -4.90541153e-02, ...,\n",
"           -4.90541153e-02, -4.90541153e-02, -4.90541153e-02],\n",
"          [-2.07840752e-02, -2.07840752e-02, -2.07840752e-02, ...,\n",
"           -2.07840752e-02, -2.07840752e-02, -2.07840752e-02],\n",
"          [-6.26482053e-02, -6.26482053e-02, -6.26482053e-02, ...,\n",
"           -6.26482053e-02, -6.26482053e-02, -6.26482053e-02],\n",
"          [-1.87769720e-02, -1.87769720e-02, -1.87769720e-02, ...,\n",
"           -1.87769720e-02, -1.87769720e-02, -1.87769720e-02]],\n",
"\n",
"         [[-2.93147925e-01, -2.93147925e-01, -1.37007380e+00, ...,\n",
"           -2.93147925e-01, -1.37007380e+00, -1.37007380e+00],\n",
"          [-5.96352945e-01, -5.96352945e-01, -5.96352945e-01, ...,\n",
"           -8.00324538e-01, -8.00324538e-01, -8.00324538e-01],\n",
"          [-2.34994269e-01, -2.34994269e-01, -1.56339141e+00, ...,\n",
"           -2.34994269e-01, -1.56339141e+00, -2.34994269e-01],\n",
"          [-6.43727660e-01, -6.43727660e-01, -6.43727660e-01, ...,\n",
"           -7.45136541e-01, -6.43727660e-01, -7.45136541e-01]],\n",
"\n",
"         ...,\n",
"\n",
"         [[-2.83110205e-01, -1.40013673e+00, -2.83110205e-01, ...,\n",
"           -2.83110205e-01, -1.40013673e+00, -2.83110205e-01],\n",
"          [-8.22611297e-01, -8.22611297e-01, -8.22611297e-01, ...,\n",
"           -5.78539059e-01, -5.78539059e-01, -8.22611297e-01],\n",
"          [-1.59522608e+00, -2.26728293e-01, -2.26728293e-01, ...,\n",
"           -1.59522608e+00, -2.26728293e-01, -2.26728293e-01],\n",
"          [-6.24893450e-01, -6.24893450e-01, -6.24893450e-01, ...,\n",
"           -7.66402976e-01, -7.66402976e-01, -7.66402976e-01]],\n",
"\n",
"         [[-9.00744882e-01, -9.00744882e-01, -5.21325431e-01, ...,\n",
"           -5.21325431e-01, -9.00744882e-01, -5.21325431e-01],\n",
"          [-2.51400937e-01, -2.51400937e-01, -1.50377467e+00, ...,\n",
"           -2.51400937e-01, -2.51400937e-01, -2.51400937e-01],\n",
"          [-7.59166221e-01, -7.59166221e-01, -7.59166221e-01, ...,\n",
"           -6.31218036e-01, -6.31218036e-01, -7.59166221e-01],\n",
"          [-2.29494293e-01, -2.29494293e-01, -1.58443074e+00, ...,\n",
"           -1.58443074e+00, -2.29494293e-01, -2.29494293e-01]],\n",
"\n",
"         [[-2.27401061e-01, -1.59258685e+00, -2.27401061e-01, ...,\n",
"           -2.27401061e-01, -1.59258685e+00, -2.27401061e-01],\n",
"          [-1.01343482e-01, -1.01343482e-01, -2.33948356e+00, ...,\n",
"           -1.01343482e-01, -1.01343482e-01, -1.01343482e-01],\n",
"          [-2.83927841e-01, -2.83927841e-01, -2.83927841e-01, ...,\n",
"           -2.83927841e-01, -2.83927841e-01, -2.83927841e-01],\n",
"          [-9.19043538e-02, -9.19043538e-02, -9.19043538e-02, ...,\n",
"           -9.19043538e-02, -9.19043538e-02, -9.19043538e-02]]],\n",
"\n",
"\n",
"        [[[-4.56259368e-01, -1.00416466e+00, -4.56259368e-01, ...,\n",
"           -4.56259368e-01, -4.56259368e-01, -4.56259368e-01],\n",
"          [-7.05154843e-01, -6.81281993e-01, -6.81281993e-01, ...,\n",
"           -6.81281993e-01, -6.81281993e-01, -6.81281993e-01],\n",
"          [-1.18448781e+00, -3.65143300e-01, -3.65143300e-01, ...,\n",
"           -3.65143300e-01, -3.65143300e-01, -3.65143300e-01],\n",
"          [-8.23245162e-01, -8.23245162e-01, -8.23245162e-01, ...,\n",
"           -8.23245162e-01, -8.23245162e-01, -5.78042749e-01]],\n",
"\n",
"         [[-4.20254312e-02, -4.20254312e-02, -4.20254312e-02, ...,\n",
"           -4.20254312e-02, -4.20254312e-02, -4.20254312e-02],\n",
"          [-2.39409111e-02, -2.39409111e-02, -2.39409111e-02, ...,\n",
"           -2.39409111e-02, -2.39409111e-02, -2.39409111e-02],\n",
"          [-5.47778814e-02, -5.47778814e-02, -5.47778814e-02, ...,\n",
"           -5.47778814e-02, -5.47778814e-02, -5.47778814e-02],\n",
"          [-1.92328870e-02, -1.92328870e-02, -1.92328870e-02, ...,\n",
"           -1.92328870e-02, -1.92328870e-02, -1.92328870e-02]],\n",
"\n",
"         [[-3.73956444e-01, -3.73956444e-01, -1.16477414e+00, ...,\n",
"           -3.73956444e-01, -1.16477414e+00, -1.16477414e+00],\n",
"          [-5.89612747e-01, -5.89612747e-01, -5.89612747e-01, ...,\n",
"           -8.08652293e-01, -8.08652293e-01, -8.08652293e-01],\n",
"          [-2.96895579e-01, -2.96895579e-01, -1.35915248e+00, ...,\n",
"           -2.96895579e-01, -1.35915248e+00, -2.96895579e-01],\n",
"          [-6.94292844e-01, -6.94292844e-01, -6.94292844e-01, ...,\n",
"           -6.92002828e-01, -6.94292844e-01, -6.92002828e-01]],\n",
"\n",
"         ...,\n",
"\n",
"         [[-3.34815718e-01, -1.25691631e+00, -3.34815718e-01, ...,\n",
"           -3.34815718e-01, -1.25691631e+00, -3.34815718e-01],\n",
"          [-8.83571321e-01, -8.83571321e-01, -8.83571321e-01, ...,\n",
"           -5.33248874e-01, -5.33248874e-01, -8.83571321e-01],\n",
"          [-1.45832136e+00, -2.64781559e-01, -2.64781559e-01, ...,\n",
"           -1.45832136e+00, -2.64781559e-01, -2.64781559e-01],\n",
"          [-6.30729194e-01, -6.30729194e-01, -6.30729194e-01, ...,\n",
"           -7.59722077e-01, -7.59722077e-01, -7.59722077e-01]],\n",
"\n",
"         [[-9.47813610e-01, -9.47813610e-01, -4.90349229e-01, ...,\n",
"           -4.90349229e-01, -9.47813610e-01, -4.90349229e-01],\n",
"          [-3.05480845e-01, -3.05480845e-01, -1.33472337e+00, ...,\n",
"           -3.05480845e-01, -3.05480845e-01, -3.05480845e-01],\n",
"          [-7.90479214e-01, -7.90479214e-01, -7.90479214e-01, ...,\n",
"           -6.04454042e-01, -6.04454042e-01, -7.90479214e-01],\n",
"          [-2.51801705e-01, -2.51801705e-01, -1.50237380e+00, ...,\n",
"           -1.50237380e+00, -2.51801705e-01, -2.51801705e-01]],\n",
"\n",
"         [[-1.56983275e-01, -1.92908104e+00, -1.56983275e-01, ...,\n",
"           -1.56983275e-01, -1.92908104e+00, -1.56983275e-01],\n",
"          [-9.16261328e-02, -9.16261328e-02, -2.43550204e+00, ...,\n",
"           -9.16261328e-02, -9.16261328e-02, -9.16261328e-02],\n",
"          [-2.01293174e-01, -2.01293174e-01, -2.01293174e-01, ...,\n",
"           -2.01293174e-01, -2.01293174e-01, -2.01293174e-01],\n",
"          [-7.40896805e-02, -7.40896805e-02, -7.40896805e-02, ...,\n",
"           -7.40896805e-02, -7.40896805e-02, -7.40896805e-02]]],\n",
"\n",
"\n",
"        [[[-4.84051769e-01, -9.57845538e-01, -4.84051769e-01, ...,\n",
"           -4.84051769e-01, -4.84051769e-01, -4.84051769e-01],\n",
"          [-9.57931672e-01, -4.83998143e-01, -4.83998143e-01, ...,\n",
"           -4.83998143e-01, -4.83998143e-01, -4.83998143e-01],\n",
"          [-1.34321374e+00, -3.02464820e-01, -3.02464820e-01, ...,\n",
"           -3.02464820e-01, -3.02464820e-01, -3.02464820e-01],\n",
"          [-8.13629622e-01, -8.13629622e-01, -8.13629622e-01, ...,\n",
"           -8.13629622e-01, -8.13629622e-01, -5.85632374e-01]],\n",
"\n",
"         [[-3.59508381e-02, -3.59508381e-02, -3.59508381e-02, ...,\n",
"           -3.59508381e-02, -3.59508381e-02, -3.59508381e-02],\n",
"          [-1.40891014e-02, -1.40891014e-02, -1.40891014e-02, ...,\n",
"           -1.40891014e-02, -1.40891014e-02, -1.40891014e-02],\n",
"          [-6.25337867e-02, -6.25337867e-02, -6.25337867e-02, ...,\n",
"           -6.25337867e-02, -6.25337867e-02, -6.25337867e-02],\n",
"          [-1.79822659e-02, -1.79822659e-02, -1.79822659e-02, ...,\n",
"           -1.79822659e-02, -1.79822659e-02, -1.79822659e-02]],\n",
"\n",
"         [[-3.10890992e-01, -3.10890992e-01, -1.31973445e+00, ...,\n",
"           -3.10890992e-01, -1.31973445e+00, -1.31973445e+00],\n",
"          [-6.63055924e-01, -6.63055924e-01, -6.63055924e-01, ...,\n",
"           -7.24172088e-01, -7.24172088e-01, -7.24172088e-01],\n",
"          [-1.88007278e-01, -1.88007278e-01, -1.76380589e+00, ...,\n",
"           -1.88007278e-01, -1.76380589e+00, -1.88007278e-01],\n",
"          [-5.51360100e-01, -5.51360100e-01, -5.51360100e-01, ...,\n",
"           -8.58412546e-01, -5.51360100e-01, -8.58412546e-01]],\n",
"\n",
"         ...,\n",
"\n",
"         [[-3.54186008e-01, -1.20980453e+00, -3.54186008e-01, ...,\n",
"           -3.54186008e-01, -1.20980453e+00, -3.54186008e-01],\n",
"          [-6.48152920e-01, -6.48152920e-01, -6.48152920e-01, ...,\n",
"           -7.40261699e-01, -7.40261699e-01, -6.48152920e-01],\n",
"          [-1.63856589e+00, -2.15992220e-01, -2.15992220e-01, ...,\n",
"           -1.63856589e+00, -2.15992220e-01, -2.15992220e-01],\n",
"          [-6.19188380e-01, -6.19188380e-01, -6.19188380e-01, ...,\n",
"           -7.73015883e-01, -7.73015883e-01, -7.73015883e-01]],\n",
"\n",
"         [[-1.39651026e+00, -1.39651026e+00, -2.84299834e-01, ...,\n",
"           -2.84299834e-01, -1.39651026e+00, -2.84299834e-01],\n",
"          [-1.19969007e-01, -1.19969007e-01, -2.17990673e+00, ...,\n",
"           -1.19969007e-01, -1.19969007e-01, -1.19969007e-01],\n",
"          [-1.00248629e+00, -1.00248629e+00, -1.00248629e+00, ...,\n",
"           -4.57231022e-01, -4.57231022e-01, -1.00248629e+00],\n",
"          [-1.51003383e-01, -1.51003383e-01, -1.96500482e+00, ...,\n",
"           -1.96500482e+00, -1.51003383e-01, -1.51003383e-01]],\n",
"\n",
"         [[-1.62326302e-01, -1.89821224e+00, -1.62326302e-01, ...,\n",
"           -1.62326302e-01, -1.89821224e+00, -1.62326302e-01],\n",
"          [-6.60834517e-02, -6.60834517e-02, -2.74969669e+00, ...,\n",
"           -6.60834517e-02, -6.60834517e-02, -6.60834517e-02],\n",
"          [-2.70560558e-01, -2.70560558e-01, -2.70560558e-01, ...,\n",
"           -2.70560558e-01, -2.70560558e-01, -2.70560558e-01],\n",
"          [-8.37560189e-02, -8.37560189e-02, -8.37560189e-02, ...,\n",
"           -8.37560189e-02, -8.37560189e-02, -8.37560189e-02]]],\n",
"\n",
"\n",
"        ...,\n",
"\n",
"\n",
"        [[[-5.29224736e-01, -8.89311639e-01, -5.29224736e-01, ...,\n",
"           -5.29224736e-01, -5.29224736e-01, -5.29224736e-01],\n",
"          [-6.37757888e-01, -7.51785274e-01, -7.51785274e-01, ...,\n",
"           -7.51785274e-01, -7.51785274e-01, -7.51785274e-01],\n",
"          [-1.43887609e+00, -2.70751794e-01, -2.70751794e-01, ...,\n",
"           -2.70751794e-01, -2.70751794e-01, -2.70751794e-01],\n",
"          [-4.61115603e-01, -4.61115603e-01, -4.61115603e-01, ...,\n",
"           -4.61115603e-01, -4.61115603e-01, -9.95820465e-01]],\n",
"\n",
"         [[-2.04877782e-02, -2.04877782e-02, -2.04877782e-02, ...,\n",
"           -2.04877782e-02, -2.04877782e-02, -2.04877782e-02],\n",
"          [-1.60545512e-02, -1.60545512e-02, -1.60545512e-02, ...,\n",
"           -1.60545512e-02, -1.60545512e-02, -1.60545512e-02],\n",
"          [-4.53924246e-02, -4.53924246e-02, -4.53924246e-02, ...,\n",
"           -4.53924246e-02, -4.53924246e-02, -4.53924246e-02],\n",
"          [-2.43494301e-02, -2.43494301e-02, -2.43494301e-02, ...,\n",
"           -2.43494301e-02, -2.43494301e-02, -2.43494301e-02]],\n",
"\n",
"         [[-7.21957859e-01, -7.21957859e-01, -6.65143365e-01, ...,\n",
"           -7.21957859e-01, -6.65143365e-01, -6.65143365e-01],\n",
"          [-8.56007208e-01, -8.56007208e-01, -8.56007208e-01, ...,\n",
"           -5.53133197e-01, -5.53133197e-01, -5.53133197e-01],\n",
"          [-3.86478837e-01, -3.86478837e-01, -1.13770174e+00, ...,\n",
"           -3.86478837e-01, -1.13770174e+00, -3.86478837e-01],\n",
"          [-6.35979153e-01, -6.35979153e-01, -6.35979153e-01, ...,\n",
"           -7.53782618e-01, -6.35979153e-01, -7.53782618e-01]],\n",
"\n",
"         ...,\n",
"\n",
"         [[-4.01593767e-01, -1.10640023e+00, -4.01593767e-01, ...,\n",
"           -4.01593767e-01, -1.10640023e+00, -4.01593767e-01],\n",
"          [-9.48599262e-01, -9.48599262e-01, -9.48599262e-01, ...,\n",
"           -4.89852319e-01, -4.89852319e-01, -9.48599262e-01],\n",
"          [-1.71192635e+00, -1.99082500e-01, -1.99082500e-01, ...,\n",
"           -1.71192635e+00, -1.99082500e-01, -1.99082500e-01],\n",
"          [-3.47144882e-01, -3.47144882e-01, -3.47144882e-01, ...,\n",
"           -1.22656930e+00, -1.22656930e+00, -1.22656930e+00]],\n",
"\n",
"         [[-1.54090820e+00, -1.54090820e+00, -2.41035777e-01, ...,\n",
"           -2.41035777e-01, -1.54090820e+00, -2.41035777e-01],\n",
"          [-1.93189922e-01, -1.93189922e-01, -1.73912187e+00, ...,\n",
"           -1.93189922e-01, -1.93189922e-01, -1.93189922e-01],\n",
"          [-9.69002464e-01, -9.69002464e-01, -9.69002464e-01, ...,\n",
"           -4.77167429e-01, -4.77167429e-01, -9.69002464e-01],\n",
"          [-2.81087776e-01, -2.81087776e-01, -1.40634224e+00, ...,\n",
"           -1.40634224e+00, -2.81087776e-01, -2.81087776e-01]],\n",
"\n",
"         [[-1.71489868e-01, -1.84775096e+00, -1.71489868e-01, ...,\n",
"           -1.71489868e-01, -1.84775096e+00, -1.71489868e-01],\n",
"          [-1.36510696e-01, -1.36510696e-01, -2.05883131e+00, ...,\n",
"           -1.36510696e-01, -1.36510696e-01, -1.36510696e-01],\n",
"          [-3.50443041e-01, -3.50443041e-01, -3.50443041e-01, ...,\n",
"           -3.50443041e-01, -3.50443041e-01, -3.50443041e-01],\n",
"          [-2.01113729e-01, -2.01113729e-01, -2.01113729e-01, ...,\n",
"           -2.01113729e-01, -2.01113729e-01, -2.01113729e-01]]],\n",
"\n",
"\n",
"        [[[-3.26755077e-01, -1.27747718e+00, -3.26755077e-01, ...,\n",
"           -3.26755077e-01, -3.26755077e-01, -3.26755077e-01],\n",
"          [-6.75270493e-01, -7.11349270e-01, -7.11349270e-01, ...,\n",
"           -7.11349270e-01, -7.11349270e-01, -7.11349270e-01],\n",
"          [-1.14427761e+00, -3.83391359e-01, -3.83391359e-01, ...,\n",
"           -3.83391359e-01, -3.83391359e-01, -3.83391359e-01],\n",
"          [-6.65774998e-01, -6.65774998e-01, -6.65774998e-01, ...,\n",
"           -6.65774998e-01, -6.65774998e-01, -7.21289736e-01]],\n",
"\n",
"         [[-1.34907758e-02, -1.34907758e-02, -1.34907758e-02, ...,\n",
"           -1.34907758e-02, -1.34907758e-02, -1.34907758e-02],\n",
"          [-5.42707905e-03, -5.42707905e-03, -5.42707905e-03, ...,\n",
"           -5.42707905e-03, -5.42707905e-03, -5.42707905e-03],\n",
"          [-1.11711274e-02, -1.11711274e-02, -1.11711274e-02, ...,\n",
"           -1.11711274e-02, -1.11711274e-02, -1.11711274e-02],\n",
"          [-5.53329713e-03, -5.53329713e-03, -5.53329713e-03, ...,\n",
"           -5.53329713e-03, -5.53329713e-03, -5.53329713e-03]],\n",
"\n",
"         [[-2.37144613e-01, -2.37144613e-01, -1.55531532e+00, ...,\n",
"           -2.37144613e-01, -1.55531532e+00, -1.55531532e+00],\n",
"          [-5.11601165e-01, -5.11601165e-01, -5.11601165e-01, ...,\n",
"           -9.15128547e-01, -9.15128547e-01, -9.15128547e-01],\n",
"          [-2.80333841e-01, -2.80333841e-01, -1.40866870e+00, ...,\n",
"           -2.80333841e-01, -1.40866870e+00, -2.80333841e-01],\n",
"          [-5.03863029e-01, -5.03863029e-01, -5.03863029e-01, ...,\n",
"           -9.26826372e-01, -5.03863029e-01, -9.26826372e-01]],\n",
"\n",
"         ...,\n",
"\n",
"         [[-4.99049730e-01, -9.34218736e-01, -4.99049730e-01, ...,\n",
"           -4.99049730e-01, -9.34218736e-01, -4.99049730e-01],\n",
"          [-4.81875851e-01, -4.81875851e-01, -4.81875851e-01, ...,\n",
"           -9.61350167e-01, -9.61350167e-01, -4.81875851e-01],\n",
"          [-8.23318513e-01, -5.77985353e-01, -5.77985353e-01, ...,\n",
"           -8.23318513e-01, -5.77985353e-01, -5.77985353e-01],\n",
"          [-9.49390730e-01, -9.49390730e-01, -9.49390730e-01, ...,\n",
"           -4.89352375e-01, -4.89352375e-01, -4.89352375e-01]],\n",
"\n",
"         [[-5.94121073e-01, -5.94121073e-01, -8.03068218e-01, ...,\n",
"           -8.03068218e-01, -5.94121073e-01, -8.03068218e-01],\n",
"          [-4.01299583e-01, -4.01299583e-01, -1.10699576e+00, ...,\n",
"           -4.01299583e-01, -4.01299583e-01, -4.01299583e-01],\n",
"          [-6.83637185e-01, -6.83637185e-01, -6.83637185e-01, ...,\n",
"           -7.02748485e-01, -7.02748485e-01, -6.83637185e-01],\n",
"          [-4.07766037e-01, -4.07766037e-01, -1.09402625e+00, ...,\n",
"           -1.09402625e+00, -4.07766037e-01, -4.07766037e-01]],\n",
"\n",
"         [[-1.97745039e-01, -1.71802052e+00, -1.97745039e-01, ...,\n",
"           -1.97745039e-01, -1.71802052e+00, -1.97745039e-01],\n",
"          [-8.39779027e-02, -8.39779027e-02, -2.51889670e+00, ...,\n",
"           -8.39779027e-02, -8.39779027e-02, -8.39779027e-02],\n",
"          [-1.66230842e-01, -1.66230842e-01, -1.66230842e-01, ...,\n",
"           -1.66230842e-01, -1.66230842e-01, -1.66230842e-01],\n",
"          [-8.55575034e-02, -8.55575034e-02, -8.55575034e-02, ...,\n",
"           -8.55575034e-02, -8.55575034e-02, -8.55575034e-02]]],\n",
"\n",
"\n",
"        [[[-3.99039569e-01, -1.11158857e+00, -3.99039569e-01, ...,\n",
"           -3.99039569e-01, -3.99039569e-01, -3.99039569e-01],\n",
"          [-8.34917198e-01, -5.69002924e-01, -5.69002924e-01, ...,\n",
"           -5.69002924e-01, -5.69002924e-01, -5.69002924e-01],\n",
"          [-9.77518622e-01, -4.71995282e-01, -4.71995282e-01, ...,\n",
"           -4.71995282e-01, -4.71995282e-01, -4.71995282e-01],\n",
"          [-8.11743897e-01, -8.11743897e-01, -8.11743897e-01, ...,\n",
"           -8.11743897e-01, -8.11743897e-01, -5.87136197e-01]],\n",
"\n",
"         [[-3.06403024e-02, -3.06403024e-02, -3.06403024e-02, ...,\n",
"           -3.06403024e-02, -3.06403024e-02, -3.06403024e-02],\n",
"          [-1.16277293e-02, -1.16277293e-02, -1.16277293e-02, ...,\n",
"           -1.16277293e-02, -1.16277293e-02, -1.16277293e-02],\n",
"          [-2.49814387e-02, -2.49814387e-02, -2.49814387e-02, ...,\n",
"           -2.49814387e-02, -2.49814387e-02, -2.49814387e-02],\n",
"          [-1.21151302e-02, -1.21151302e-02, -1.21151302e-02, ...,\n",
"           -1.21151302e-02, -1.21151302e-02, -1.21151302e-02]],\n",
"\n",
"         [[-2.87441963e-01, -2.87441963e-01, -1.38701503e+00, ...,\n",
"           -2.87441963e-01, -1.38701503e+00, -1.38701503e+00],\n",
"          [-6.34424612e-01, -6.34424612e-01, -6.34424612e-01, ...,\n",
"           -7.55534406e-01, -7.55534406e-01, -7.55534406e-01],\n",
"          [-3.43313763e-01, -3.43313763e-01, -1.23586117e+00, ...,\n",
"           -3.43313763e-01, -1.23586117e+00, -3.43313763e-01],\n",
"          [-6.15232763e-01, -6.15232763e-01, -6.15232763e-01, ...,\n",
"           -7.77649131e-01, -6.15232763e-01, -7.77649131e-01]],\n",
"\n",
"         ...,\n",
"\n",
"         [[-3.59344118e-01, -1.19777230e+00, -3.59344118e-01, ...,\n",
"           -3.59344118e-01, -1.19777230e+00, -3.59344118e-01],\n",
"          [-6.25578860e-01, -6.25578860e-01, -6.25578860e-01, ...,\n",
"           -7.65613958e-01, -7.65613958e-01, -6.25578860e-01],\n",
"          [-1.05787564e+00, -4.26473124e-01, -4.26473124e-01, ...,\n",
"           -1.05787564e+00, -4.26473124e-01, -4.26473124e-01],\n",
"          [-7.43729363e-01, -7.43729363e-01, -7.43729363e-01, ...,\n",
"           -6.45000839e-01, -6.45000839e-01, -6.45000839e-01]],\n",
"\n",
"         [[-7.15665131e-01, -7.15665131e-01, -6.71125142e-01, ...,\n",
"           -6.71125142e-01, -7.15665131e-01, -6.71125142e-01],\n",
"          [-3.07126990e-01, -3.07126990e-01, -1.33013026e+00, ...,\n",
"           -3.07126990e-01, -3.07126990e-01, -3.07126990e-01],\n",
"          [-8.26819895e-01, -8.26819895e-01, -8.26819895e-01, ...,\n",
"           -5.75254242e-01, -5.75254242e-01, -8.26819895e-01],\n",
"          [-3.18217231e-01, -3.18217231e-01, -1.29991392e+00, ...,\n",
"           -1.29991392e+00, -3.18217231e-01, -3.18217231e-01]],\n",
"\n",
"         [[-1.61945154e-01, -1.90037761e+00, -1.61945154e-01, ...,\n",
"           -1.61945154e-01, -1.90037761e+00, -1.61945154e-01],\n",
"          [-6.39879615e-02, -6.39879615e-02, -2.78088370e+00, ...,\n",
"           -6.39879615e-02, -6.39879615e-02, -6.39879615e-02],\n",
"          [-1.33587907e-01, -1.33587907e-01, -1.33587907e-01, ...,\n",
"           -1.33587907e-01, -1.33587907e-01, -1.33587907e-01],\n",
"          [-6.65985007e-02, -6.65985007e-02, -6.65985007e-02, ...,\n",
"           -6.65985007e-02, -6.65985007e-02, -6.65985007e-02]]]],\n",
"\n",
"\n",
"\n",
"       [[[[-4.48057726e-01, -1.01851119e+00, -4.48057726e-01, ...,\n",
"           -4.48057726e-01, -4.48057726e-01, -4.48057726e-01],\n",
"          [-6.23757076e-01, -7.67713694e-01, -7.67713694e-01, ...,\n",
"           -7.67713694e-01, -7.67713694e-01, -7.67713694e-01],\n",
"          [-1.34967403e+00, -3.00193049e-01, -3.00193049e-01, ...,\n",
"           -3.00193049e-01, -3.00193049e-01, -3.00193049e-01],\n",
"          [-5.93879084e-01, -5.93879084e-01, -5.93879084e-01, ...,\n",
"           -5.93879084e-01, -5.93879084e-01, -8.03366521e-01]],\n",
"\n",
"         [[-6.98777815e-03, -6.98777815e-03, -6.98777815e-03, ...,\n",
"           -6.98777815e-03, -6.98777815e-03, -6.98777815e-03],\n",
"          [-4.56709485e-03, -4.56709485e-03, -4.56709485e-03, ...,\n",
"           -4.56709485e-03, -4.56709485e-03, -4.56709485e-03],\n",
"          [-1.12576957e-02, -1.12576957e-02, -1.12576957e-02, ...,\n",
"           -1.12576957e-02, -1.12576957e-02, -1.12576957e-02],\n",
"          [-4.87565144e-03, -4.87565144e-03, -4.87565144e-03, ...,\n",
"           -4.87565144e-03, -4.87565144e-03, -4.87565144e-03]],\n",
"\n",
"         [[-4.48060848e-01, -4.48060848e-01, -1.01850567e+00, ...,\n",
"           -4.48060848e-01, -1.01850567e+00, -1.01850567e+00],\n",
"          [-6.23761088e-01, -6.23761088e-01, -6.23761088e-01, ...,\n",
"           -7.67709062e-01, -7.67709062e-01, -7.67709062e-01],\n",
"          [-3.00195290e-01, -3.00195290e-01, -1.34966763e+00, ...,\n",
"           -3.00195290e-01, -1.34966763e+00, -3.00195290e-01],\n",
"          [-5.93882955e-01, -5.93882955e-01, -5.93882955e-01, ...,\n",
"           -8.03361748e-01, -5.93882955e-01, -8.03361748e-01]],\n",
"\n",
"         ...,\n",
"\n",
"         [[-4.42251951e-01, -1.02886530e+00, -4.42251951e-01, ...,\n",
"           -4.42251951e-01, -1.02886530e+00, -4.42251951e-01],\n",
"          [-7.76406676e-01, -7.76406676e-01, -7.76406676e-01, ...,\n",
"           -6.16290174e-01, -6.16290174e-01, -7.76406676e-01],\n",
"          [-1.36166827e+00, -2.96027405e-01, -2.96027405e-01, ...,\n",
"           -1.36166827e+00, -2.96027405e-01, -2.96027405e-01],\n",
"          [-5.86674652e-01, -5.86674652e-01, -5.86674652e-01, ...,\n",
"           -8.12321973e-01, -8.12321973e-01, -8.12321973e-01]],\n",
"\n",
"         [[-8.95943726e-01, -8.95943726e-01, -5.24623997e-01, ...,\n",
"           -5.24623997e-01, -8.95943726e-01, -5.24623997e-01],\n",
"          [-3.71778093e-01, -3.71778093e-01, -1.16959467e+00, ...,\n",
"           -3.71778093e-01, -3.71778093e-01, -3.71778093e-01],\n",
"          [-6.40742709e-01, -6.40742709e-01, -6.40742709e-01, ...,\n",
"           -7.48450493e-01, -7.48450493e-01, -6.40742709e-01],\n",
"          [-3.92588371e-01, -3.92588371e-01, -1.12487413e+00, ...,\n",
"           -1.12487413e+00, -3.92588371e-01, -3.92588371e-01]],\n",
"\n",
"         [[-2.09529472e-01, -1.66582700e+00, -2.09529472e-01, ...,\n",
"           -2.09529472e-01, -1.66582700e+00, -2.09529472e-01],\n",
"          [-1.41642179e-01, -1.41642179e-01, -2.02443656e+00, ...,\n",
"           -1.41642179e-01, -1.41642179e-01, -1.41642179e-01],\n",
"          [-3.19425689e-01, -3.19425689e-01, -3.19425689e-01, ...,\n",
"           -3.19425689e-01, -3.19425689e-01, -3.19425689e-01],\n",
"          [-1.50546821e-01, -1.50546821e-01, -1.50546821e-01, ...,\n",
"           -1.50546821e-01, -1.50546821e-01, -1.50546821e-01]]],\n",
"\n",
"\n",
"        [[[-5.54722368e-01, -8.53859894e-01, -5.54722368e-01, ...,\n",
"           -5.54722368e-01, -5.54722368e-01, -5.54722368e-01],\n",
"          [-6.13003730e-01, -7.80277630e-01, -7.80277630e-01, ...,\n",
"           -7.80277630e-01, -7.80277630e-01, -7.80277630e-01],\n",
"          [-1.30155936e+00, -3.17601424e-01, -3.17601424e-01, ...,\n",
"           -3.17601424e-01, -3.17601424e-01, -3.17601424e-01],\n",
"          [-6.59104751e-01, -6.59104751e-01, -6.59104751e-01, ...,\n",
"           -6.59104751e-01, -6.59104751e-01, -7.28389463e-01]],\n",
"\n",
"         [[-5.17415943e-02, -5.17415943e-02, -5.17415943e-02, ...,\n",
"           -5.17415943e-02, -5.17415943e-02, -5.17415943e-02],\n",
"          [-4.54925334e-02, -4.54925334e-02, -4.54925334e-02, ...,\n",
"           -4.54925334e-02, -4.54925334e-02, -4.54925334e-02],\n",
"          [-1.00140738e-01, -1.00140738e-01, -1.00140738e-01, ...,\n",
"           -1.00140738e-01, -1.00140738e-01, -1.00140738e-01],\n",
"          [-4.13327014e-02, -4.13327014e-02, -4.13327014e-02, ...,\n",
"           -4.13327014e-02, -4.13327014e-02, -4.13327014e-02]],\n",
"\n",
"         [[-3.92590798e-01, -3.92590798e-01, -1.12486908e+00, ...,\n",
"           -3.92590798e-01, -1.12486908e+00, -1.12486908e+00],\n",
"          [-4.37341039e-01, -4.37341039e-01, -4.37341039e-01, ...,\n",
"           -1.03775570e+00, -1.03775570e+00, -1.03775570e+00],\n",
"          [-2.17058129e-01, -2.17058129e-01, -1.63415683e+00, ...,\n",
"           -2.17058129e-01, -1.63415683e+00, -2.17058129e-01],\n",
"          [-4.73161962e-01, -4.73161962e-01, -4.73161962e-01, ...,\n",
"           -9.75587432e-01, -4.73161962e-01, -9.75587432e-01]],\n",
"\n",
"         ...,\n",
"\n",
"         [[-5.68938172e-01, -8.35001682e-01, -5.68938172e-01, ...,\n",
"           -5.68938172e-01, -8.35001682e-01, -5.68938172e-01],\n",
"          [-7.62496637e-01, -7.62496637e-01, -7.62496637e-01, ...,\n",
"           -6.28296752e-01, -6.28296752e-01, -7.62496637e-01],\n",
"          [-1.27759390e+00, -3.26709973e-01, -3.26709973e-01, ...,\n",
"           -1.27759390e+00, -3.26709973e-01, -3.26709973e-01],\n",
"          [-6.75205725e-01, -6.75205725e-01, -6.75205725e-01, ...,\n",
"           -7.11416421e-01, -7.11416421e-01, -7.11416421e-01]],\n",
"\n",
"         [[-1.15537728e+00, -1.15537728e+00, -3.78246946e-01, ...,\n",
"           -3.78246946e-01, -1.15537728e+00, -3.78246946e-01],\n",
"          [-3.38562467e-01, -3.38562467e-01, -1.24755643e+00, ...,\n",
"           -3.38562467e-01, -3.38562467e-01, -3.38562467e-01],\n",
"          [-7.40366904e-01, -7.40366904e-01, -7.40366904e-01, ...,\n",
"           -6.48056981e-01, -6.48056981e-01, -7.40366904e-01],\n",
"          [-3.11388418e-01, -3.11388418e-01, -1.31837157e+00, ...,\n",
"           -1.31837157e+00, -3.11388418e-01, -3.11388418e-01]],\n",
"\n",
"         [[-6.83764414e-02, -2.71672036e+00, -6.83764414e-02, ...,\n",
"           -6.83764414e-02, -2.71672036e+00, -6.83764414e-02],\n",
"          [-6.01780590e-02, -6.01780590e-02, -2.84038560e+00, ...,\n",
"           -6.01780590e-02, -6.01780590e-02, -6.01780590e-02],\n",
"          [-1.31346614e-01, -1.31346614e-01, -1.31346614e-01, ...,\n",
"           -1.31346614e-01, -1.31346614e-01, -1.31346614e-01],\n",
"          [-5.47117210e-02, -5.47117210e-02, -5.47117210e-02, ...,\n",
"           -5.47117210e-02, -5.47117210e-02, -5.47117210e-02]]],\n",
"\n",
"\n",
"        [[[-5.09129187e-01, -9.18840795e-01, -5.09129187e-01, ...,\n",
"           -5.09129187e-01, -5.09129187e-01, -5.09129187e-01],\n",
"          [-9.32671931e-01, -5.00052030e-01, -5.00052030e-01, ...,\n",
"           -5.00052030e-01, -5.00052030e-01, -5.00052030e-01],\n",
"          [-1.03194343e+00, -4.40543962e-01, -4.40543962e-01, ...,\n",
"           -4.40543962e-01, -4.40543962e-01, -4.40543962e-01],\n",
"          [-7.67371130e-01, -7.67371130e-01, -7.67371130e-01, ...,\n",
"           -7.67371130e-01, -7.67371130e-01, -6.24053806e-01]],\n",
"\n",
"         [[-3.28008253e-02, -3.28008253e-02, -3.28008253e-02, ...,\n",
"           -3.28008253e-02, -3.28008253e-02, -3.28008253e-02],\n",
"          [-1.42595826e-02, -1.42595826e-02, -1.42595826e-02, ...,\n",
"           -1.42595826e-02, -1.42595826e-02, -1.42595826e-02],\n",
"          [-3.92094581e-02, -3.92094581e-02, -3.92094581e-02, ...,\n",
"           -3.92094581e-02, -3.92094581e-02, -3.92094581e-02],\n",
"          [-1.89984185e-02, -1.89984185e-02, -1.89984185e-02, ...,\n",
"           -1.89984185e-02, -1.89984185e-02, -1.89984185e-02]],\n",
"\n",
"         [[-5.09545497e-01, -5.09545497e-01, -9.18213999e-01, ...,\n",
"           -5.09545497e-01, -9.18213999e-01, -9.18213999e-01],\n",
"          [-9.33304704e-01, -9.33304704e-01, -9.33304704e-01, ...,\n",
"           -4.99641697e-01, -4.99641697e-01, -4.99641697e-01],\n",
"          [-4.40915760e-01, -4.40915760e-01, -1.03127213e+00, ...,\n",
"           -4.40915760e-01, -1.03127213e+00, -4.40915760e-01],\n",
"          [-7.67930128e-01, -7.67930128e-01, -7.67930128e-01, ...,\n",
"           -6.23569698e-01, -7.67930128e-01, -6.23569698e-01]],\n",
"\n",
"         ...,\n",
"\n",
"         [[-3.18258299e-01, -1.29980432e+00, -3.18258299e-01, ...,\n",
"           -3.18258299e-01, -1.29980432e+00, -3.18258299e-01],\n",
"          [-7.65175070e-01, -7.65175070e-01, -7.65175070e-01, ...,\n",
"           -6.25960554e-01, -6.25960554e-01, -7.65175070e-01],\n",
"          [-1.43514782e+00, -2.71913935e-01, -2.71913935e-01, ...,\n",
"           -1.43514782e+00, -2.71913935e-01, -2.71913935e-01],\n",
"          [-5.01668498e-01, -5.01668498e-01, -5.01668498e-01, ...,\n",
"           -9.30185591e-01, -9.30185591e-01, -9.30185591e-01]],\n",
"\n",
"         [[-7.42355284e-01, -7.42355284e-01, -6.46247372e-01, ...,\n",
"           -6.46247372e-01, -7.42355284e-01, -6.46247372e-01],\n",
"          [-3.30193938e-01, -3.30193938e-01, -1.26863336e+00, ...,\n",
"           -3.30193938e-01, -3.30193938e-01, -3.30193938e-01],\n",
"          [-6.51272417e-01, -6.51272417e-01, -6.51272417e-01, ...,\n",
"           -7.36852368e-01, -7.36852368e-01, -6.51272417e-01],\n",
"          [-4.20351473e-01, -4.20351473e-01, -1.06948832e+00, ...,\n",
"           -1.06948832e+00, -4.20351473e-01, -4.20351473e-01]],\n",
"\n",
"         [[-2.84545106e-01, -1.39576474e+00, -2.84545106e-01, ...,\n",
"           -2.84545106e-01, -1.39576474e+00, -2.84545106e-01],\n",
"          [-1.32579455e-01, -1.32579455e-01, -2.08613060e+00, ...,\n",
"           -1.32579455e-01, -1.32579455e-01, -1.32579455e-01],\n",
"          [-3.32706940e-01, -3.32706940e-01, -3.32706940e-01, ...,\n",
"           -3.32706940e-01, -3.32706940e-01, -3.32706940e-01],\n",
"          [-1.73392542e-01, -1.73392542e-01, -1.73392542e-01, ...,\n",
"           -1.73392542e-01, -1.73392542e-01, -1.73392542e-01]]],\n",
"\n",
"\n",
"        ...,\n",
"\n",
"\n",
"        [[[-5.35976703e-01, -8.79711833e-01, -5.35976703e-01, ...,\n",
"           -5.35976703e-01, -5.35976703e-01, -5.35976703e-01],\n",
"          [-8.78282662e-01, -5.36991391e-01, -5.36991391e-01, ...,\n",
"           -5.36991391e-01, -5.36991391e-01, -5.36991391e-01],\n",
"          [-9.92800266e-01, -4.62889210e-01, -4.62889210e-01, ...,\n",
"           -4.62889210e-01, -4.62889210e-01, -4.62889210e-01],\n",
"          [-7.87392545e-01, -7.87392545e-01, -7.87392545e-01, ...,\n",
"           -7.87392545e-01, -7.87392545e-01, -6.07024017e-01]],\n",
"\n",
"         [[-1.12958285e-02, -1.12958285e-02, -1.12958285e-02, ...,\n",
"           -1.12958285e-02, -1.12958285e-02, -1.12958285e-02],\n",
"          [-5.70991428e-03, -5.70991428e-03, -5.70991428e-03, ...,\n",
"           -5.70991428e-03, -5.70991428e-03, -5.70991428e-03],\n",
"          [-1.35916972e-02, -1.35916972e-02, -1.35916972e-02, ...,\n",
"           -1.35916972e-02, -1.35916972e-02, -1.35916972e-02],\n",
"          [-6.70349779e-03, -6.70349779e-03, -6.70349779e-03, ...,\n",
"           -6.70349779e-03, -6.70349779e-03, -6.70349779e-03]],\n",
"\n",
"         [[-5.63657933e-01, -5.63657933e-01, -8.41933625e-01, ...,\n",
"           -5.63657933e-01, -8.41933625e-01, -8.41933625e-01],\n",
"          [-9.17062290e-01, -9.17062290e-01, -9.17062290e-01, ...,\n",
"           -5.10311582e-01, -5.10311582e-01, -5.10311582e-01],\n",
"          [-4.87646853e-01, -4.87646853e-01, -9.52098473e-01, ...,\n",
"           -4.87646853e-01, -9.52098473e-01, -4.87646853e-01],\n",
"          [-8.23596142e-01, -8.23596142e-01, -8.23596142e-01, ...,\n",
"           -5.77768177e-01, -8.23596142e-01, -5.77768177e-01]],\n",
"\n",
"         ...,\n",
"\n",
"         [[-5.78394663e-01, -8.22795639e-01, -5.78394663e-01, ...,\n",
"           -5.78394663e-01, -8.22795639e-01, -5.78394663e-01],\n",
"          [-4.96909493e-01, -4.96909493e-01, -4.96909493e-01, ...,\n",
"           -9.37534916e-01, -9.37534916e-01, -4.96909493e-01],\n",
"          [-9.31433353e-01, -5.00856449e-01, -5.00856449e-01, ...,\n",
"           -9.31433353e-01, -5.00856449e-01, -5.00856449e-01],\n",
"          [-8.42746008e-01, -8.42746008e-01, -8.42746008e-01, ...,\n",
"           -5.63043326e-01, -5.63043326e-01, -5.63043326e-01]],\n",
"\n",
"         [[-9.74818292e-01, -9.74818292e-01, -4.73627626e-01, ...,\n",
"           -4.73627626e-01, -9.74818292e-01, -4.73627626e-01],\n",
"          [-2.66489814e-01, -2.66489814e-01, -1.45270688e+00, ...,\n",
"           -2.66489814e-01, -2.66489814e-01, -2.66489814e-01],\n",
"          [-8.63007884e-01, -8.63007884e-01, -8.63007884e-01, ...,\n",
"           -5.47993144e-01, -5.47993144e-01, -8.63007884e-01],\n",
"          [-3.06521647e-01, -3.06521647e-01, -1.33181597e+00, ...,\n",
"           -1.33181597e+00, -3.06521647e-01, -3.06521647e-01]],\n",
"\n",
"         [[-2.02597281e-01, -1.69612410e+00, -2.02597281e-01, ...,\n",
"           -2.02597281e-01, -1.69612410e+00, -2.02597281e-01],\n",
"          [-1.07243418e-01, -1.07243418e-01, -2.28579663e+00, ...,\n",
"           -1.07243418e-01, -1.07243418e-01, -1.07243418e-01],\n",
"          [-2.39438692e-01, -2.39438692e-01, -2.39438692e-01, ...,\n",
"           -2.39438692e-01, -2.39438692e-01, -2.39438692e-01],\n",
"          [-1.24842652e-01, -1.24842652e-01, -1.24842652e-01, ...,\n",
"           -1.24842652e-01, -1.24842652e-01, -1.24842652e-01]]],\n",
"\n",
"\n",
"        [[[-3.94608627e-01, -1.12068537e+00, -3.94608627e-01, ...,\n",
"           -3.94608627e-01, -3.94608627e-01, -3.94608627e-01],\n",
"          [-8.20034825e-01, -5.80562194e-01, -5.80562194e-01, ...,\n",
"           -5.80562194e-01, -5.80562194e-01, -5.80562194e-01],\n",
"          [-1.09158710e+00, -4.08996298e-01, -4.08996298e-01, ...,\n",
"           -4.08996298e-01, -4.08996298e-01, -4.08996298e-01],\n",
"          [-5.86668498e-01, -5.86668498e-01, -5.86668498e-01, ...,\n",
"           -5.86668498e-01, -5.86668498e-01, -8.12329686e-01]],\n",
"\n",
"         [[-9.12151044e-03, -9.12151044e-03, -9.12151044e-03, ...,\n",
"           -9.12151044e-03, -9.12151044e-03, -9.12151044e-03],\n",
"          [-3.48304958e-03, -3.48304958e-03, -3.48304958e-03, ...,\n",
"           -3.48304958e-03, -3.48304958e-03, -3.48304958e-03],\n",
"          [-8.73504407e-03, -8.73504407e-03, -8.73504407e-03, ...,\n",
"           -8.73504407e-03, -8.73504407e-03, -8.73504407e-03],\n",
"          [-5.54010109e-03, -5.54010109e-03, -5.54010109e-03, ...,\n",
"           -5.54010109e-03, -5.54010109e-03, -5.54010109e-03]],\n",
"\n",
"         [[-4.07185589e-01, -4.07185589e-01, -1.09518018e+00, ...,\n",
"           -4.07185589e-01, -1.09518018e+00, -1.09518018e+00],\n",
"          [-8.41523401e-01, -8.41523401e-01, -8.41523401e-01, ...,\n",
"           -5.63968621e-01, -5.63968621e-01, -5.63968621e-01],\n",
"          [-4.21942210e-01, -4.21942210e-01, -1.06645086e+00, ...,\n",
"           -4.21942210e-01, -1.06645086e+00, -4.21942210e-01],\n",
"          [-6.03749465e-01, -6.03749465e-01, -6.03749465e-01, ...,\n",
"           -7.91328504e-01, -6.03749465e-01, -7.91328504e-01]],\n",
"\n",
"         ...,\n",
"\n",
"         [[-3.25616235e-01, -1.28043005e+00, -3.25616235e-01, ...,\n",
"           -3.25616235e-01, -1.28043005e+00, -3.25616235e-01],\n",
"          [-6.87793808e-01, -6.87793808e-01, -6.87793808e-01, ...,\n",
"           -6.98529366e-01, -6.98529366e-01, -6.87793808e-01],\n",
"          [-1.24922059e+00, -3.37892711e-01, -3.37892711e-01, ...,\n",
"           -1.24922059e+00, -3.37892711e-01, -3.37892711e-01],\n",
"          [-4.91538734e-01, -4.91538734e-01, -4.91538734e-01, ...,\n",
"           -9.45936996e-01, -9.45936996e-01, -9.45936996e-01]],\n",
"\n",
"         [[-8.58759056e-01, -8.58759056e-01, -5.51105281e-01, ...,\n",
"           -5.51105281e-01, -8.58759056e-01, -5.51105281e-01],\n",
"          [-2.46808136e-01, -2.46808136e-01, -1.52001128e+00, ...,\n",
"           -2.46808136e-01, -2.46808136e-01, -2.46808136e-01],\n",
"          [-8.84050890e-01, -8.84050890e-01, -8.84050890e-01, ...,\n",
"           -5.32911174e-01, -5.32911174e-01, -8.84050890e-01],\n",
"          [-3.68605925e-01, -3.68605925e-01, -1.17667525e+00, ...,\n",
"           -1.17667525e+00, -3.68605925e-01, -3.68605925e-01]],\n",
"\n",
"         [[-1.55970807e-01, -1.93505841e+00, -1.55970807e-01, ...,\n",
"           -1.55970807e-01, -1.93505841e+00, -1.55970807e-01],\n",
"          [-6.22906484e-02, -6.22906484e-02, -2.80692763e+00, ...,\n",
"           -6.22906484e-02, -6.22906484e-02, -6.22906484e-02],\n",
"          [-1.49806377e-01, -1.49806377e-01, -1.49806377e-01, ...,\n",
"           -1.49806377e-01, -1.49806377e-01, -1.49806377e-01],\n",
"          [-9.74306919e-02, -9.74306919e-02, -9.74306919e-02, ...,\n",
"           -9.74306919e-02, -9.74306919e-02, -9.74306919e-02]]],\n",
"\n",
"\n",
"        [[[-5.78383750e-01, -8.22809573e-01, -5.78383750e-01, ...,\n",
"           -5.78383750e-01, -5.78383750e-01, -5.78383750e-01],\n",
"          [-6.11222364e-01, -7.82387445e-01, -7.82387445e-01, ...,\n",
"           -7.82387445e-01, -7.82387445e-01, -7.82387445e-01],\n",
"          [-1.28666998e+00, -3.23224926e-01, -3.23224926e-01, ...,\n",
"           -3.23224926e-01, -3.23224926e-01, -3.23224926e-01],\n",
"          [-7.25410463e-01, -7.25410463e-01, -7.25410463e-01, ...,\n",
"           -7.25410463e-01, -7.25410463e-01, -6.61892365e-01]],\n",
"\n",
"         [[-3.56713513e-02, -3.56713513e-02, -3.56713513e-02, ...,\n",
"           -3.56713513e-02, -3.56713513e-02, -3.56713513e-02],\n",
"          [-3.31928177e-02, -3.31928177e-02, -3.31928177e-02, ...,\n",
"           -3.31928177e-02, -3.31928177e-02, -3.31928177e-02],\n",
"          [-7.18871076e-02, -7.18871076e-02, -7.18871076e-02, ...,\n",
"           -7.18871076e-02, -7.18871076e-02, -7.18871076e-02],\n",
"          [-2.63401313e-02, -2.63401313e-02, -2.63401313e-02, ...,\n",
"           -2.63401313e-02, -2.63401313e-02, -2.63401313e-02]],\n",
"\n",
"         [[-3.31638177e-01, -3.31638177e-01, -1.26495135e+00, ...,\n",
"           -3.31638177e-01, -1.26495135e+00, -1.26495135e+00],\n",
"          [-3.52865624e-01, -3.52865624e-01, -3.52865624e-01, ...,\n",
"           -1.21291806e+00, -1.21291806e+00, -1.21291806e+00],\n",
"          [-1.75299033e-01, -1.75299033e-01, -1.82763144e+00, ...,\n",
"           -1.75299033e-01, -1.82763144e+00, -1.75299033e-01],\n",
"          [-4.28571955e-01, -4.28571955e-01, -4.28571955e-01, ...,\n",
"           -1.05394121e+00, -4.28571955e-01, -1.05394121e+00]],\n",
"\n",
"         ...,\n",
"\n",
"         [[-6.69397379e-01, -7.17474786e-01, -6.69397379e-01, ...,\n",
"           -6.69397379e-01, -7.17474786e-01, -6.69397379e-01],\n",
"          [-6.80634786e-01, -6.80634786e-01, -6.80634786e-01, ...,\n",
"           -7.05818121e-01, -7.05818121e-01, -6.80634786e-01],\n",
"          [-1.14851446e+00, -3.81417820e-01, -3.81417820e-01, ...,\n",
"           -1.14851446e+00, -3.81417820e-01, -3.81417820e-01],\n",
"          [-8.31498118e-01, -8.31498118e-01, -8.31498118e-01, ...,\n",
"           -5.71631604e-01, -5.71631604e-01, -5.71631604e-01]],\n",
"\n",
"         [[-9.79444184e-01, -9.79444184e-01, -4.70835592e-01, ...,\n",
"           -4.70835592e-01, -9.79444184e-01, -4.70835592e-01],\n",
"          [-4.43950107e-01, -4.43950107e-01, -1.02581944e+00, ...,\n",
"           -4.43950107e-01, -4.43950107e-01, -4.43950107e-01],\n",
"          [-5.93465761e-01, -5.93465761e-01, -5.93465761e-01, ...,\n",
"           -8.03876402e-01, -8.03876402e-01, -5.93465761e-01],\n",
"          [-3.65998207e-01, -3.65998207e-01, -1.18255072e+00, ...,\n",
"           -1.18255072e+00, -3.65998207e-01, -3.65998207e-01]],\n",
"\n",
"         [[-1.24951205e-01, -2.14165713e+00, -1.24951205e-01, ...,\n",
"           -1.24951205e-01, -2.14165713e+00, -1.24951205e-01],\n",
"          [-1.16619033e-01, -1.16619033e-01, -2.20658570e+00, ...,\n",
"           -1.16619033e-01, -1.16619033e-01, -1.16619033e-01],\n",
"          [-2.41504333e-01, -2.41504333e-01, -2.41504333e-01, ...,\n",
"           -2.41504333e-01, -2.41504333e-01, -2.41504333e-01],\n",
"          [-9.33242902e-02, -9.33242902e-02, -9.33242902e-02, ...,\n",
"           -9.33242902e-02, -9.33242902e-02, -9.33242902e-02]]]],\n",
"\n",
"\n",
"\n",
"       [[[[-4.13740122e-01, -1.08226490e+00, -4.13740122e-01, ...,\n",
"           -4.13740122e-01, -4.13740122e-01, -4.13740122e-01],\n",
"          [-8.45812650e-01, -5.60731145e-01, -5.60731145e-01, ...,\n",
"           -5.60731145e-01, -5.60731145e-01, -5.60731145e-01],\n",
"          [-1.07271615e+00, -4.18669077e-01, -4.18669077e-01, ...,\n",
"           -4.18669077e-01, -4.18669077e-01, -4.18669077e-01],\n",
"          [-6.14140845e-01, -6.14140845e-01, -6.14140845e-01, ...,\n",
"           -6.14140845e-01, -6.14140845e-01, -7.78935137e-01]],\n",
"\n",
"         [[-3.35026954e-02, -3.35026954e-02, -3.35026954e-02, ...,\n",
"           -3.35026954e-02, -3.35026954e-02, -3.35026954e-02],\n",
"          [-1.30434866e-02, -1.30434866e-02, -1.30434866e-02, ...,\n",
"           -1.30434866e-02, -1.30434866e-02, -1.30434866e-02],\n",
"          [-3.30290126e-02, -3.30290126e-02, -3.30290126e-02, ...,\n",
"           -3.30290126e-02, -3.30290126e-02, -3.30290126e-02],\n",
"          [-2.03786321e-02, -2.03786321e-02, -2.03786321e-02, ...,\n",
"           -2.03786321e-02, -2.03786321e-02, -2.03786321e-02]],\n",
"\n",
"         [[-3.88139638e-01, -3.88139638e-01, -1.13419061e+00, ...,\n",
"           -3.88139638e-01, -1.13419061e+00, -1.13419061e+00],\n",
"          [-8.02300098e-01, -8.02300098e-01, -8.02300098e-01, ...,\n",
"           -5.94744789e-01, -5.94744789e-01, -5.94744789e-01],\n",
"          [-3.92819768e-01, -3.92819768e-01, -1.12439304e+00, ...,\n",
"           -3.92819768e-01, -1.12439304e+00, -3.92819768e-01],\n",
"          [-5.79308951e-01, -5.79308951e-01, -5.79308951e-01, ...,\n",
"           -8.21629439e-01, -5.79308951e-01, -8.21629439e-01]],\n",
"\n",
"         ...,\n",
"\n",
"         [[-4.69539801e-01, -9.81602778e-01, -4.69539801e-01, ...,\n",
"           -4.69539801e-01, -9.81602778e-01, -4.69539801e-01],\n",
"          [-4.96550158e-01, -4.96550158e-01, -4.96550158e-01, ...,\n",
"           -9.38093463e-01, -9.38093463e-01, -4.96550158e-01],\n",
"          [-9.72574599e-01, -4.74989328e-01, -4.74989328e-01, ...,\n",
"           -9.72574599e-01, -4.74989328e-01, -4.74989328e-01],\n",
"          [-6.88989614e-01, -6.88989614e-01, -6.88989614e-01, ...,\n",
"           -6.97322105e-01, -6.97322105e-01, -6.97322105e-01]],\n",
"\n",
"         [[-8.15969534e-01, -8.15969534e-01, -5.83773415e-01, ...,\n",
"           -5.83773415e-01, -8.15969534e-01, -5.83773415e-01],\n",
"          [-2.66586834e-01, -2.66586834e-01, -1.45238924e+00, ...,\n",
"           -2.66586834e-01, -2.66586834e-01, -2.66586834e-01],\n",
"          [-8.24070884e-01, -8.24070884e-01, -8.24070884e-01, ...,\n",
"           -5.77397059e-01, -5.77397059e-01, -8.24070884e-01],\n",
"          [-3.91407666e-01, -3.91407666e-01, -1.12733427e+00, ...,\n",
"           -1.12733427e+00, -3.91407666e-01, -3.91407666e-01]],\n",
"\n",
"         [[-2.72124593e-01, -1.43447396e+00, -2.72124593e-01, ...,\n",
"           -2.72124593e-01, -1.43447396e+00, -2.72124593e-01],\n",
"          [-1.13791133e-01, -1.13791133e-01, -2.22974678e+00, ...,\n",
"           -1.13791133e-01, -1.13791133e-01, -1.13791133e-01],\n",
"          [-2.68694387e-01, -2.68694387e-01, -2.68694387e-01, ...,\n",
"           -2.68694387e-01, -2.68694387e-01, -2.68694387e-01],\n",
"          [-1.73101238e-01, -1.73101238e-01, -1.73101238e-01, ...,\n",
"           -1.73101238e-01, -1.73101238e-01, -1.73101238e-01]]],\n",
"\n",
"\n",
"        [[[-6.24521061e-01, -7.66832145e-01, -6.24521061e-01, ...,\n",
"           -6.24521061e-01, -6.24521061e-01, -6.24521061e-01],\n",
"          [-8.79500541e-01, -5.36126561e-01, -5.36126561e-01, ...,\n",
"           -5.36126561e-01, -5.36126561e-01, -5.36126561e-01],\n",
"          [-1.21563045e+00, -3.51720116e-01, -3.51720116e-01, ...,\n",
"           -3.51720116e-01, -3.51720116e-01, -3.51720116e-01],\n",
"          [-8.08580497e-01, -8.08580497e-01, -8.08580497e-01, ...,\n",
"           -8.08580497e-01, -8.08580497e-01, -5.89670424e-01]],\n",
"\n",
"         [[-1.07796896e-02, -1.07796896e-02, -1.07796896e-02, ...,\n",
"           -1.07796896e-02, -1.07796896e-02, -1.07796896e-02],\n",
"          [-6.64622111e-03, -6.64622111e-03, -6.64622111e-03, ...,\n",
"           -6.64622111e-03, -6.64622111e-03, -6.64622111e-03],\n",
"          [-2.20565401e-02, -2.20565401e-02, -2.20565401e-02, ...,\n",
"           -2.20565401e-02, -2.20565401e-02, -2.20565401e-02],\n",
"          [-7.52381281e-03, -7.52381281e-03, -7.52381281e-03, ...,\n",
"           -7.52381281e-03, -7.52381281e-03, -7.52381281e-03]],\n",
"\n",
"         [[-3.28662143e-01, -3.28662143e-01, -1.27255931e+00, ...,\n",
"           -3.28662143e-01, -1.27255931e+00, -1.27255931e+00],\n",
"          [-4.90059488e-01, -4.90059488e-01, -4.90059488e-01, ...,\n",
"           -9.48271589e-01, -9.48271589e-01, -9.48271589e-01],\n",
"          [-1.73193998e-01, -1.73193998e-01, -1.83869041e+00, ...,\n",
"           -1.73193998e-01, -1.83869041e+00, -1.73193998e-01],\n",
"          [-4.43660987e-01, -4.43660987e-01, -4.43660987e-01, ...,\n",
"           -1.02633699e+00, -4.43660987e-01, -1.02633699e+00]],\n",
"\n",
"         ...,\n",
"\n",
"         [[-5.31528885e-01, -8.86017980e-01, -5.31528885e-01, ...,\n",
"           -5.31528885e-01, -8.86017980e-01, -5.31528885e-01],\n",
"          [-6.29699203e-01, -6.29699203e-01, -6.29699203e-01, ...,\n",
"           -7.60895171e-01, -7.60895171e-01, -6.29699203e-01],\n",
"          [-1.36944914e+00, -2.93360800e-01, -2.93360800e-01, ...,\n",
"           -1.36944914e+00, -2.93360800e-01, -2.93360800e-01],\n",
"          [-6.96518877e-01, -6.96518877e-01, -6.96518877e-01, ...,\n",
"           -6.89786815e-01, -6.89786815e-01, -6.89786815e-01]],\n",
"\n",
"         [[-6.61687263e-01, -6.61687263e-01, -7.25629063e-01, ...,\n",
"           -7.25629063e-01, -6.61687263e-01, -7.25629063e-01],\n",
"          [-5.04346124e-01, -5.04346124e-01, -9.26089388e-01, ...,\n",
"           -5.04346124e-01, -5.04346124e-01, -5.04346124e-01],\n",
"          [-3.75605712e-01, -3.75605712e-01, -3.75605712e-01, ...,\n",
"           -1.16114676e+00, -1.16114676e+00, -3.75605712e-01],\n",
"          [-5.55513939e-01, -5.55513939e-01, -8.52793297e-01, ...,\n",
"           -8.52793297e-01, -5.55513939e-01, -5.55513939e-01]],\n",
"\n",
"         [[-6.53711845e-02, -2.76018126e+00, -6.53711845e-02, ...,\n",
"           -6.53711845e-02, -2.76018126e+00, -6.53711845e-02],\n",
"          [-4.07244564e-02, -4.07244564e-02, -3.22121960e+00, ...,\n",
"           -4.07244564e-02, -4.07244564e-02, -4.07244564e-02],\n",
"          [-1.30159290e-01, -1.30159290e-01, -1.30159290e-01, ...,\n",
"           -1.30159290e-01, -1.30159290e-01, -1.30159290e-01],\n",
"          [-4.59997520e-02, -4.59997520e-02, -4.59997520e-02, ...,\n",
"           -4.59997520e-02, -4.59997520e-02, -4.59997520e-02]]],\n",
"\n",
"\n",
"        [[[-5.65937621e-01, -8.38930442e-01, -5.65937621e-01, ...,\n",
"           -5.65937621e-01, -5.65937621e-01, -5.65937621e-01],\n",
"          [-8.46682900e-01, -5.60077256e-01, -5.60077256e-01, ...,\n",
"           -5.60077256e-01, -5.60077256e-01, -5.60077256e-01],\n",
"          [-1.26894575e+00, -3.30071746e-01, -3.30071746e-01, ...,\n",
"           -3.30071746e-01, -3.30071746e-01, -3.30071746e-01],\n",
"          [-8.90473422e-01, -8.90473422e-01, -8.90473422e-01, ...,\n",
"           -8.90473422e-01, -8.90473422e-01, -5.28415057e-01]],\n",
"\n",
"         [[-1.81765257e-02, -1.81765257e-02, -1.81765257e-02, ...,\n",
"           -1.81765257e-02, -1.81765257e-02, -1.81765257e-02],\n",
"          [-1.04271857e-02, -1.04271857e-02, -1.04271857e-02, ...,\n",
"           -1.04271857e-02, -1.04271857e-02, -1.04271857e-02],\n",
"          [-3.50762680e-02, -3.50762680e-02, -3.50762680e-02, ...,\n",
"           -3.50762680e-02, -3.50762680e-02, -3.50762680e-02],\n",
"          [-9.67302801e-03, -9.67302801e-03, -9.67302801e-03, ...,\n",
"           -9.67302801e-03, -9.67302801e-03, -9.67302801e-03]],\n",
"\n",
"         [[-2.77326538e-01, -2.77326538e-01, -1.41802037e+00, ...,\n",
"           -2.77326538e-01, -1.41802037e+00, -1.41802037e+00],\n",
"          [-4.44227646e-01, -4.44227646e-01, -4.44227646e-01, ...,\n",
"           -1.02532301e+00, -1.02532301e+00, -1.02532301e+00],\n",
"          [-1.52047566e-01, -1.52047566e-01, -1.95862257e+00, ...,\n",
"           -1.52047566e-01, -1.95862257e+00, -1.52047566e-01],\n",
"          [-4.71950397e-01, -4.71950397e-01, -4.71950397e-01, ...,\n",
"           -9.77593039e-01, -4.71950397e-01, -9.77593039e-01]],\n",
"\n",
"         ...,\n",
"\n",
"         [[-3.77294547e-01, -1.15745210e+00, -3.77294547e-01, ...,\n",
"           -3.77294547e-01, -1.15745210e+00, -3.77294547e-01],\n",
"          [-8.09495228e-01, -8.09495228e-01, -8.09495228e-01, ...,\n",
"           -5.88936140e-01, -5.88936140e-01, -8.09495228e-01],\n",
"          [-1.65751555e+00, -2.11476820e-01, -2.11476820e-01, ...,\n",
"           -1.65751555e+00, -2.11476820e-01, -2.11476820e-01],\n",
"          [-6.23223673e-01, -6.23223673e-01, -6.23223673e-01, ...,\n",
"           -7.68330041e-01, -7.68330041e-01, -7.68330041e-01]],\n",
"\n",
"         [[-4.86167479e-01, -4.86167479e-01, -9.54456888e-01, ...,\n",
"           -9.54456888e-01, -4.86167479e-01, -9.54456888e-01],\n",
"          [-6.48534458e-01, -6.48534458e-01, -7.39843515e-01, ...,\n",
"           -6.48534458e-01, -6.48534458e-01, -6.48534458e-01],\n",
"          [-2.78910330e-01, -2.78910330e-01, -2.78910330e-01, ...,\n",
"           -1.41308092e+00, -1.41308092e+00, -2.78910330e-01],\n",
"          [-6.13238458e-01, -6.13238458e-01, -7.80000235e-01, ...,\n",
"           -7.80000235e-01, -6.13238458e-01, -6.13238458e-01]],\n",
"\n",
"         [[-7.15689624e-02, -2.67266485e+00, -7.15689624e-02, ...,\n",
"           -7.15689624e-02, -2.67266485e+00, -7.15689624e-02],\n",
"          [-4.15221903e-02, -4.15221903e-02, -3.20221655e+00, ...,\n",
"           -4.15221903e-02, -4.15221903e-02, -4.15221903e-02],\n",
"          [-1.34874595e-01, -1.34874595e-01, -1.34874595e-01, ...,\n",
"           -1.34874595e-01, -1.34874595e-01, -1.34874595e-01],\n",
"          [-3.85619208e-02, -3.85619208e-02, -3.85619208e-02, ...,\n",
"           -3.85619208e-02, -3.85619208e-02, -3.85619208e-02]]],\n",
"\n",
"\n",
"        ...,\n",
"\n",
"\n",
"        [[[-4.46126986e-01, -1.02193596e+00, -4.46126986e-01, ...,\n",
"           -4.46126986e-01, -4.46126986e-01, -4.46126986e-01],\n",
"          [-6.33787356e-01, -7.56254196e-01, -7.56254196e-01, ...,\n",
"           -7.56254196e-01, -7.56254196e-01, -7.56254196e-01],\n",
"          [-1.26487413e+00, -3.31668546e-01, -3.31668546e-01, ...,\n",
"           -3.31668546e-01, -3.31668546e-01, -3.31668546e-01],\n",
"          [-5.64308930e-01, -5.64308930e-01, -5.64308930e-01, ...,\n",
"           -5.64308930e-01, -5.64308930e-01, -8.41074405e-01]],\n",
"\n",
"         [[-4.55093595e-02, -4.55093595e-02, -4.55093595e-02, ...,\n",
"           -4.55093595e-02, -4.55093595e-02, -4.55093595e-02],\n",
"          [-2.91601008e-02, -2.91601008e-02, -2.91601008e-02, ...,\n",
"           -2.91601008e-02, -2.91601008e-02, -2.91601008e-02],\n",
"          [-6.44417484e-02, -6.44417484e-02, -6.44417484e-02, ...,\n",
"           -6.44417484e-02, -6.44417484e-02, -6.44417484e-02],\n",
"          [-3.39434448e-02, -3.39434448e-02, -3.39434448e-02, ...,\n",
"           -3.39434448e-02, -3.39434448e-02, -3.39434448e-02]],\n",
"\n",
"         [[-5.09597906e-01, -5.09597906e-01, -9.18135139e-01, ...,\n",
"           -5.09597906e-01, -9.18135139e-01, -9.18135139e-01],\n",
"          [-7.15800547e-01, -7.15800547e-01, -7.15800547e-01, ...,\n",
"           -6.70995642e-01, -6.70995642e-01, -6.70995642e-01],\n",
"          [-3.81786699e-01, -3.81786699e-01, -1.14772054e+00, ...,\n",
"           -3.81786699e-01, -1.14772054e+00, -3.81786699e-01],\n",
"          [-6.39898177e-01, -6.39898177e-01, -6.39898177e-01, ...,\n",
"           -7.49391907e-01, -6.39898177e-01, -7.49391907e-01]],\n",
"\n",
"         ...,\n",
"\n",
"         [[-4.76364269e-01, -9.70317275e-01, -4.76364269e-01, ...,\n",
"           -4.76364269e-01, -9.70317275e-01, -4.76364269e-01],\n",
"          [-7.13658756e-01, -7.13658756e-01, -7.13658756e-01, ...,\n",
"           -6.73047888e-01, -6.73047888e-01, -7.13658756e-01],\n",
"          [-1.20681075e+00, -3.55461141e-01, -3.55461141e-01, ...,\n",
"           -1.20681075e+00, -3.55461141e-01, -3.55461141e-01],\n",
"          [-6.00433645e-01, -6.00433645e-01, -6.00433645e-01, ...,\n",
"           -7.95343149e-01, -7.95343149e-01, -7.95343149e-01]],\n",
"\n",
"         [[-1.22089856e+00, -1.22089856e+00, -3.49507837e-01, ...,\n",
"           -3.49507837e-01, -1.22089856e+00, -3.49507837e-01],\n",
"          [-2.35762772e-01, -2.35762772e-01, -1.56049564e+00, ...,\n",
"           -2.35762772e-01, -2.35762772e-01, -2.35762772e-01],\n",
"          [-9.82810739e-01, -9.82810739e-01, -9.82810739e-01, ...,\n",
"           -4.68816619e-01, -4.68816619e-01, -9.82810739e-01],\n",
"          [-2.70204374e-01, -2.70204374e-01, -1.44063860e+00, ...,\n",
"           -1.44063860e+00, -2.70204374e-01, -2.70204374e-01]],\n",
"\n",
"         [[-8.78101394e-02, -2.47616212e+00, -8.78101394e-02, ...,\n",
"           -8.78101394e-02, -2.47616212e+00, -8.78101394e-02],\n",
"          [-5.66891421e-02, -5.66891421e-02, -2.89838326e+00, ...,\n",
"           -5.66891421e-02, -5.66891421e-02, -5.66891421e-02],\n",
"          [-1.23287951e-01, -1.23287951e-01, -1.23287951e-01, ...,\n",
"           -1.23287951e-01, -1.23287951e-01, -1.23287951e-01],\n",
"          [-6.58420154e-02, -6.58420154e-02, -6.58420154e-02, ...,\n",
"           -6.58420154e-02, -6.58420154e-02, -6.58420154e-02]]],\n",
"\n",
"\n",
"        [[[-2.87509441e-01, -1.38681243e+00, -2.87509441e-01, ...,\n",
"           -2.87509441e-01, -2.87509441e-01, -2.87509441e-01],\n",
"          [-5.27140561e-01, -8.92306805e-01, -8.92306805e-01, ...,\n",
"           -8.92306805e-01, -8.92306805e-01, -8.92306805e-01],\n",
"          [-1.49411022e+00, -2.54180536e-01, -2.54180536e-01, ...,\n",
"           -2.54180536e-01, -2.54180536e-01, -2.54180536e-01],\n",
"          [-4.78708532e-01, -4.78708532e-01, -4.78708532e-01, ...,\n",
"           -4.78708532e-01, -4.78708532e-01, -9.66487383e-01]],\n",
"\n",
"         [[-1.30613491e-02, -1.30613491e-02, -1.30613491e-02, ...,\n",
"           -1.30613491e-02, -1.30613491e-02, -1.30613491e-02],\n",
"          [-6.28969111e-03, -6.28969111e-03, -6.28969111e-03, ...,\n",
"           -6.28969111e-03, -6.28969111e-03, -6.28969111e-03],\n",
"          [-1.50188052e-02, -1.50188052e-02, -1.50188052e-02, ...,\n",
"           -1.50188052e-02, -1.50188052e-02, -1.50188052e-02],\n",
"          [-7.10724959e-03, -7.10724959e-03, -7.10724959e-03, ...,\n",
"           -7.10724959e-03, -7.10724959e-03, -7.10724959e-03]],\n",
"\n",
"         [[-3.13980796e-01, -3.13980796e-01, -1.31130956e+00, ...,\n",
"           -3.13980796e-01, -1.31130956e+00, -1.31130956e+00],\n",
"          [-5.70185051e-01, -5.70185051e-01, -5.70185051e-01, ...,\n",
"           -8.33377066e-01, -8.33377066e-01, -8.33377066e-01],\n",
"          [-2.77990442e-01, -2.77990442e-01, -1.41594590e+00, ...,\n",
"           -2.77990442e-01, -1.41594590e+00, -2.77990442e-01],\n",
"          [-5.18736263e-01, -5.18736263e-01, -5.18736263e-01, ...,\n",
"           -9.04540884e-01, -5.18736263e-01, -9.04540884e-01]],\n",
"\n",
"         ...,\n",
"\n",
"         [[-4.54676451e-01, -1.00690845e+00, -4.54676451e-01, ...,\n",
"           -4.54676451e-01, -1.00690845e+00, -4.54676451e-01],\n",
"          [-6.06325283e-01, -6.06325283e-01, -6.06325283e-01, ...,\n",
"           -7.88230033e-01, -7.88230033e-01, -6.06325283e-01],\n",
"          [-1.09841997e+00, -4.05561281e-01, -4.05561281e-01, ...,\n",
"           -1.09841997e+00, -4.05561281e-01, -4.05561281e-01],\n",
"          [-7.23232633e-01, -7.23232633e-01, -7.23232633e-01, ...,\n",
"           -6.63940489e-01, -6.63940489e-01, -6.63940489e-01]],\n",
"\n",
"         [[-7.38247151e-01, -7.38247151e-01, -6.49993745e-01, ...,\n",
"           -6.49993745e-01, -7.38247151e-01, -6.49993745e-01],\n",
"          [-3.64212613e-01, -3.64212613e-01, -1.18660277e+00, ...,\n",
"           -3.64212613e-01, -3.64212613e-01, -3.64212613e-01],\n",
"          [-6.67303368e-01, -6.67303368e-01, -6.67303368e-01, ...,\n",
"           -7.19676655e-01, -7.19676655e-01, -6.67303368e-01],\n",
"          [-4.03259867e-01, -4.03259867e-01, -1.10303741e+00, ...,\n",
"           -1.10303741e+00, -4.03259867e-01, -4.03259867e-01]],\n",
"\n",
"         [[-2.43336555e-01, -1.53251209e+00, -2.43336555e-01, ...,\n",
"           -2.43336555e-01, -1.53251209e+00, -2.43336555e-01],\n",
"          [-1.24177480e-01, -1.24177480e-01, -2.14748977e+00, ...,\n",
"           -1.24177480e-01, -1.24177480e-01, -1.24177480e-01],\n",
"          [-2.75429762e-01, -2.75429762e-01, -2.75429762e-01, ...,\n",
"           -2.75429762e-01, -2.75429762e-01, -2.75429762e-01],\n",
"          [-1.39295752e-01, -1.39295752e-01, -1.39295752e-01, ...,\n",
"           -1.39295752e-01, -1.39295752e-01, -1.39295752e-01]]],\n",
"\n",
"\n",
"        [[[-4.91354503e-01, -9.46227270e-01, -4.91354503e-01, ...,\n",
"           -4.91354503e-01, -4.91354503e-01, -4.91354503e-01],\n",
"          [-8.16170853e-01, -5.83613840e-01, -5.83613840e-01, ...,\n",
"           -5.83613840e-01, -5.83613840e-01, -5.83613840e-01],\n",
"          [-1.15247046e+00, -3.79586123e-01, -3.79586123e-01, ...,\n",
"           -3.79586123e-01, -3.79586123e-01, -3.79586123e-01],\n",
"          [-6.82774436e-01, -6.82774436e-01, -6.82774436e-01, ...,\n",
"           -6.82774436e-01, -6.82774436e-01, -7.03628647e-01]],\n",
"\n",
"         [[-5.93189866e-02, -5.93189866e-02, -5.93189866e-02, ...,\n",
"           -5.93189866e-02, -5.93189866e-02, -5.93189866e-02],\n",
"          [-3.02692682e-02, -3.02692682e-02, -3.02692682e-02, ...,\n",
"           -3.02692682e-02, -3.02692682e-02, -3.02692682e-02],\n",
"          [-8.06525024e-02, -8.06525024e-02, -8.06525024e-02, ...,\n",
"           -8.06525024e-02, -8.06525024e-02, -8.06525024e-02],\n",
"          [-3.88317525e-02, -3.88317525e-02, -3.88317525e-02, ...,\n",
"           -3.88317525e-02, -3.88317525e-02, -3.88317525e-02]],\n",
"\n",
"         [[-4.10444307e-01, -4.10444307e-01, -1.08872766e+00, ...,\n",
"           -4.10444307e-01, -1.08872766e+00, -1.08872766e+00],\n",
"          [-6.97730853e-01, -6.97730853e-01, -6.97730853e-01, ...,\n",
"           -6.88584423e-01, -6.88584423e-01, -6.88584423e-01],\n",
"          [-3.14259488e-01, -3.14259488e-01, -1.31055440e+00, ...,\n",
"           -3.14259488e-01, -1.31055440e+00, -3.14259488e-01],\n",
"          [-5.78454477e-01, -5.78454477e-01, -5.78454477e-01, ...,\n",
"           -8.22719271e-01, -5.78454477e-01, -8.22719271e-01]],\n",
"\n",
"         ...,\n",
"\n",
"         [[-4.67861581e-01, -9.84409561e-01, -4.67861581e-01, ...,\n",
"           -4.67861581e-01, -9.84409561e-01, -4.67861581e-01],\n",
"          [-6.11351922e-01, -6.11351922e-01, -6.11351922e-01, ...,\n",
"           -7.82233722e-01, -7.82233722e-01, -6.11351922e-01],\n",
"          [-1.19507305e+00, -3.60513505e-01, -3.60513505e-01, ...,\n",
"           -1.19507305e+00, -3.60513505e-01, -3.60513505e-01],\n",
"          [-6.52733615e-01, -6.52733615e-01, -6.52733615e-01, ...,\n",
"           -7.35263039e-01, -7.35263039e-01, -7.35263039e-01]],\n",
"\n",
"         [[-1.08422633e+00, -1.08422633e+00, -4.12736448e-01, ...,\n",
"           -4.12736448e-01, -1.08422633e+00, -4.12736448e-01],\n",
"          [-2.28678770e-01, -2.28678770e-01, -1.58759843e+00, ...,\n",
"           -2.28678770e-01, -2.28678770e-01, -2.28678770e-01],\n",
"          [-8.85424063e-01, -8.85424063e-01, -8.85424063e-01, ...,\n",
"           -5.31945748e-01, -5.31945748e-01, -8.85424063e-01],\n",
"          [-2.85962489e-01, -2.85962489e-01, -1.39147093e+00, ...,\n",
"           -1.39147093e+00, -2.85962489e-01, -2.85962489e-01]],\n",
"\n",
"         [[-9.22478147e-02, -2.42904605e+00, -9.22478147e-02, ...,\n",
"           -9.22478147e-02, -2.42904605e+00, -9.22478147e-02],\n",
"          [-4.74514797e-02, -4.74514797e-02, -3.07167949e+00, ...,\n",
"           -4.74514797e-02, -4.74514797e-02, -4.74514797e-02],\n",
"          [-1.24707025e-01, -1.24707025e-01, -1.24707025e-01, ...,\n",
"           -1.24707025e-01, -1.24707025e-01, -1.24707025e-01],\n",
"          [-6.07289401e-02, -6.07289401e-02, -6.07289401e-02, ...,\n",
"           -6.07289401e-02, -6.07289401e-02, -6.07289401e-02]]]]])
• created_at :
2020-06-04T22:45:38.960981
arviz_version :
0.8.3
inference_library :
pymc3
inference_library_version :
3.8
" ], "text/plain": [ "\n", "Dimensions: (actor: 7, chain: 4, draw: 500, repetition: 18, treatment: 4)\n", "Coordinates:\n", " * chain (chain) int64 0 1 2 3\n", " * draw (draw) int64 0 1 2 3 4 5 6 7 ... 493 494 495 496 497 498 499\n", " * actor (actor) int64 0 1 2 3 4 5 6\n", " * treatment (treatment) int64 0 1 2 3\n", " * repetition (repetition) int64 0 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17\n", "Data variables:\n", " pulled_left (chain, draw, actor, treatment, repetition) float64 -0.5873 ... -0.06073\n", "Attributes:\n", " created_at: 2020-06-04T22:45:38.960981\n", " arviz_version: 0.8.3\n", " inference_library: pymc3\n", " inference_library_version: 3.8" ] }, "execution_count": 26, "metadata": {}, "output_type": "execute_result" } ], "source": [ "idata_11_4.log_likelihood" ] }, { "cell_type": "code", "execution_count": null, "metadata": {}, "outputs": [], "source": [] } ], "metadata": { "kernelspec": { "display_name": "Python 3", "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.6.9" } }, "nbformat": 4, "nbformat_minor": 2 }