Notebook on nbviewer (original) (raw)

• perf_counter_diff
  (chain, draw)
  float64
  0.0003155 0.0006459 ... 0.0007621
  array([[0.00031552, 0.00064593, 0.00069063, ..., 0.00062145, 0.00059205,
  0.00058207],
  [0.00159006, 0.00066053, 0.00061815, ..., 0.00068065, 0.00059797,
  0.00064247],
  [0.00075648, 0.00060866, 0.00059738, ..., 0.00059973, 0.00069401,
  0.00082627],
  [0.00060619, 0.00102361, 0.00117493, ..., 0.00123848, 0.00119367,
  0.00076208]])
• process_time_diff
  (chain, draw)
  float64
  0.0003156 0.0006464 ... 0.0007625
  array([[0.0003156 , 0.00064642, 0.00069033, ..., 0.00062148, 0.00059208,
  0.00058217],
  [0.00159045, 0.00066069, 0.00061837, ..., 0.00068108, 0.00059787,
  0.00064259],
  [0.00075682, 0.00060879, 0.00059751, ..., 0.00059983, 0.00069433,
  0.00082669],
  [0.00060628, 0.00102389, 0.00117498, ..., 0.00123882, 0.00119386,
  0.00076253]])
• index_in_trajectory
  (chain, draw)
  int64
  2 -4 6 6 4 0 -5 ... 10 -5 9 -3 -2 2
  array([[ 2, -4, 6, ..., 3, -2, 4],
  [ 2, 5, 1, ..., 1, 4, -4],
  [-2, -2, 4, ..., 2, 5, 3],
  [ 4, 4, -4, ..., -3, -2, 2]])
• tree_depth
  (chain, draw)
  int64
  2 3 3 3 3 2 3 3 ... 3 3 4 4 4 4 4 3
  array([[2, 3, 3, ..., 3, 3, 3],
  [4, 3, 3, ..., 3, 3, 3],
  [3, 3, 3, ..., 3, 3, 3],
  [3, 4, 4, ..., 4, 4, 3]])
• lp
  (chain, draw)
  float64
  -21.4 -20.69 -17.1 ... -17.56 -18.2
  array([[-21.40304775, -20.68634085, -17.09770967, ..., -16.61778863,
  -19.90588875, -18.35484707],
  [-16.43637671, -17.83211478, -17.56335636, ..., -16.65891692,
  -19.59915102, -16.16390328],
  [-16.92452321, -16.9315486 , -19.21426331, ..., -23.53971843,
  -17.57950491, -17.63383671],
  [-16.32975137, -19.40369695, -18.83500204, ..., -17.92099412,
  -17.55736677, -18.19697182]])
• diverging
  (chain, draw)
  bool
  False False False ... False False
  array([[False, False, False, ..., False, False, False],
  [False, False, False, ..., False, False, False],
  [False, False, False, ..., False, False, False],
  [False, False, False, ..., False, False, False]])
• smallest_eigval
  (chain, draw)
  float64
  nan nan nan nan ... nan nan nan nan
  array([[nan, nan, nan, ..., nan, nan, nan],
  [nan, nan, nan, ..., nan, nan, nan],
  [nan, nan, nan, ..., nan, nan, nan],
  [nan, nan, nan, ..., nan, nan, nan]])
• energy_error
  (chain, draw)
  float64
  -0.04507 -0.04706 ... 0.1187
  array([[-0.04506733, -0.04705797, -0.14389229, ..., -0.11368085,
  0.15052472, -0.14802306],
  [-0.21284673, 0.08662963, -0.00510208, ..., -0.05682078,
  0.33748444, -0.25833276],
  [-0.01412885, -0.00254213, 0.01623483, ..., 0.06621467,
  -0.17240469, 0.07795121],
  [ 0.11642224, 0.06369934, 0.03483888, ..., -0.253348 ,
  -0.01218285, 0.11874154]])
• step_size_bar
  (chain, draw)
  float64
  0.4662 0.4662 ... 0.3553 0.3553
  array([[0.46615199, 0.46615199, 0.46615199, ..., 0.46615199, 0.46615199,
  0.46615199],
  [0.41681342, 0.41681342, 0.41681342, ..., 0.41681342, 0.41681342,
  0.41681342],
  [0.42515651, 0.42515651, 0.42515651, ..., 0.42515651, 0.42515651,
  0.42515651],
  [0.35533979, 0.35533979, 0.35533979, ..., 0.35533979, 0.35533979,
  0.35533979]])
• perf_counter_start
  (chain, draw)
  float64
  7.286e+05 7.286e+05 ... 7.286e+05
  array([[728577.637351 , 728577.63774557, 728577.63849577, ...,
  728581.16186798, 728581.16256292, 728581.16322677],
  [728577.70053449, 728577.70223228, 728577.70297527, ...,
  728581.81716345, 728581.81792648, 728581.81860002],
  [728577.69719806, 728577.69804757, 728577.69873327, ...,
  728581.24639168, 728581.24706437, 728581.24785382],
  [728577.77024483, 728577.77093347, 728577.77204651, ...,
  728582.74501243, 728582.746335 , 728582.74760917]])
• reached_max_treedepth
  (chain, draw)
  bool
  False False False ... False False
  array([[False, False, False, ..., False, False, False],
  [False, False, False, ..., False, False, False],
  [False, False, False, ..., False, False, False],
  [False, False, False, ..., False, False, False]])
• max_energy_error
  (chain, draw)
  float64
  -0.1459 -0.2902 ... 0.4472 0.293
  array([[-0.14589256, -0.29019037, -0.34978825, ..., 0.11458972,
  2.25242137, -0.2532052 ],
  [-0.22106833, 0.12267941, -0.10849476, ..., 0.10262767,
  0.33748444, -0.25833276],
  [-0.11142988, 0.08886613, 0.09981523, ..., -0.15323593,
  -1.64090375, 0.07795121],
  [ 0.12336939, 0.1199349 , 0.50687014, ..., 1.472127 ,
  0.44716448, 0.29303868]])
• energy
  (chain, draw)
  float64
  23.61 24.48 22.11 ... 19.03 19.11
  array([[23.61103767, 24.47955343, 22.11011983, ..., 20.63578335,
  20.99935337, 20.85484695],
  [20.72800171, 19.00278783, 18.52529905, ..., 19.16719243,
  20.35298105, 19.98874213],
  [18.22314287, 19.18871882, 20.06453835, ..., 23.85414275,
  24.86726751, 19.36885018],
  [17.16335271, 20.70101765, 20.97146667, ..., 20.43214727,
  19.0291882 , 19.10814306]])
• acceptance_rate
  (chain, draw)
  float64
  1.0 0.9896 0.9965 ... 0.9346 0.8853
  array([[1. , 0.98962646, 0.99651247, ..., 0.96365264, 0.57939049,
  0.99439214],
  [0.99946586, 0.94441399, 1. , ..., 0.95996917, 0.78041781,
  0.9861433 ],
  [0.98691282, 0.98011764, 0.9461868 , ..., 0.96815379, 0.99290192,
  0.97650599],
  [0.9185277 , 0.9277961 , 0.940619 , ..., 0.89305968, 0.93462463,
  0.88527164]])
• largest_eigval
  (chain, draw)
  float64
  nan nan nan nan ... nan nan nan nan
  array([[nan, nan, nan, ..., nan, nan, nan],
  [nan, nan, nan, ..., nan, nan, nan],
  [nan, nan, nan, ..., nan, nan, nan],
  [nan, nan, nan, ..., nan, nan, nan]])
• n_steps
  (chain, draw)
  float64
  3.0 7.0 7.0 7.0 ... 15.0 15.0 7.0
  array([[ 3., 7., 7., ..., 7., 7., 7.],
  [15., 7., 7., ..., 7., 7., 7.],
  [ 7., 7., 7., ..., 7., 7., 7.],
  [ 7., 11., 15., ..., 15., 15., 7.]])
• step_size
  (chain, draw)
  float64
  0.5166 0.5166 ... 0.3903 0.3903
  array([[0.51662592, 0.51662592, 0.51662592, ..., 0.51662592, 0.51662592,
  0.51662592],
  [0.5857316 , 0.5857316 , 0.5857316 , ..., 0.5857316 , 0.5857316 ,
  0.5857316 ],
  [0.52728411, 0.52728411, 0.52728411, ..., 0.52728411, 0.52728411,
  0.52728411],
  [0.39029824, 0.39029824, 0.39029824, ..., 0.39029824, 0.39029824,
  0.39029824]])