extract().to_dataframe() из StanFit4Model не работает на pystan 2.18

Я оценил модель, используя pystan:

import pystan

stan_model = pystan.StanModel('stan_codes/stan_code_1.stan')
samples = stan_model.sampling(data = sdata, iter = 10, chains = 1, seed = 42)
model_results = {'mdata' : model_data, 'sdata': sdata, 'samples' : samples, 'model': stan_model}

Позже я хочу извлечь "материал" из samples в dataframe формат. я использую

mdata, sdata, samples, model = [x for x in model_results.values()]
samples.extract().to_dataframe()

Но я получаю ошибку:

 AttributeError: 'collections.OrderedDict' object has no attribute 'to_dataframe'

Согласно документации samples.extract() должен иметь атрибут to_dataframe(), право? Я делаю что-то здесь не так? я использую pystan version 2.18.0.0

1 ответ

Решение

Вы пробовали извлечение в DataFrame непосредственно из подходящего объекта

Что-то вроде...

stan_model = pystan.StanModel('stan_codes/stan_code_1.stan')
samples = stan_model.sampling(data = sdata, iter = 10, chains = 1, seed = 42)
chains = samples.to_dataframe('parameter name')
Другие вопросы по тегам