pyro-ppl/pyro

[FR] get mcmc.summary() as DataFrame

Open

#3013 aperta il 26 gen 2022

Vedi su GitHub
 (6 commenti) (1 reazione) (0 assegnatari)Python (981 fork)batch import
enhancementhelp wantedusability

Metriche repository

Star
 (8211 star)
Metriche merge PR
 (Merge medio 10g 19h) (1 PR mergiata in 30 g)

Descrizione

First of all thanks for the great package!

Issue Description

One of the useful features of the MCMC class is the MCMC.summary() method which prints summary statistics of the sampled chain.

In this issue I'm asking for the feature to be able to get this table as a Pandas DataFrame.

Code Snippet

def model(data):
    ...

nuts_kernel = NUTS(model)
mcmc = MCMC(nuts_kernel, num_samples=500)
mcmc.run(data)
mcmc.summary()

It would be very helpful if there was a simple way to get mcmc.summary as DataFrame or perhaps alternatively as a dict which can become a DataFrame.

Guida contributor