Hi everyone, is there a way to generate a pair plot where (1) the divergences are highlighted for a specific chain and a specific coordinate? I can generate plots with two of those features. But not all three :frowning: Here’s what I want to work: az.plot_pair(trace, divergences=True, …

Use .sel directly on the inferencedata instead of the coords argument. I haven’t checked the source but I guess nobody expected this usecase and consequently coords are only applied to the posterior group (or whatever group the user choses) not the sample_stats one. az.plot_pair(trace.sel(chain=0, …

Pair plots with specific chains, coords, and divergences

Questions version agnostic

daniel-saunders-phil July 23, 2024, 11:52pm 4

Ah thanks so much Oriol, I didn’t realize you could .sel on the top level of inference data. Super convenient that you can though!

Topic		Replies	Views
Combining dimensions in Arviz plots version agnostic arviz	11	1933	February 18, 2021
How do I pull a subset of an ARVIZ array within the inference data? v5 arviz	10	1832	May 17, 2022
Selecting arbitrary subsets of 2d random variables for plotting version agnostic arviz	1	306	March 14, 2023
Best way to plot and do PPC with variable that has (too) many levels Questions	7	2787	September 25, 2020
How to forestplot in arviz with only a subset of trace variables and samples Questions	4	2002	July 30, 2019

Pair plots with specific chains, coords, and divergences

Related topics