pyro-ppl/pyro

[FR] get mcmc.summary() as DataFrame

Open

#3,013 建立於 2022年1月26日

在 GitHub 查看
 (6 留言) (1 反應) (0 負責人)Python (981 fork)batch import
enhancementhelp wantedusability

倉庫指標

Star
 (8,211 star)
PR 合併指標
 (平均合併 10天 19小時) (30 天內合併 1 個 PR)

描述

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.

貢獻者指南