1

John Kruschke の本「Doing Bayesian Data Analysis」(第 2 版) の例を再現しようとしています。この例は、反復測定による単純な線形回帰に関する第 16 章からのものです。この場合、尤度関数を指定することに夢中になっていると思います。John Kruschke のサイトからデータをダウンロードし、それを使用して csv を作成しました。これは、本で彼の R コードに従っているコードです。

import pandas as pd
import pymc3 as pm
import numpy as np

data = pd.read_csv('Feldmandata.csv')
data = data[np.isfinite(data['retention'])]
data = data[data['group'] == 1]

data['log10_retention'] = np.log10(data['retention'])
nrow, ncol = data.shape[0], data.shape[1]

data['subjID'] = data['subjID']-1
subjID = data['subjID']
nsubj = len(np.unique(subjID))

y = data['time'].astype('float')
x = data['log10_retention'].astype('float')

std_y = (y - np.mean(y)) / np.std(y)
std_x = (x - np.mean(x)) / np.std(x)

sample = 1000
burnin = 500

with pm.Model() as model:
    # Priors
    m = pm.Gamma('m', 1, 0.1)
    d = pm.Gamma('d', 1, 0.1)
    G_r = pm.Deterministic('G_r', m / d**2)
    G_s = pm.Deterministic('G_s', m**2 / d**2)

    t_1G = pm.Gamma('t_1G', 0.1, 0.1)
    mu_1G = pm.Normal('mu_1G', 0, 0.1)
    t_0G = pm.Gamma('t_0G', 0.1, 0.1)
    mu_0G = pm.Gamma('mu_0G', 0, 0.01)

    b0 = pm.Normal('b0', mu = mu_0G, tau = t_0G, shape = nsubj)
    b1 = pm.Normal('b1', mu = mu_1G, tau = t_1G, shape = nsubj)
    tau = pm.Gamma('tau', G_s, G_r, shape = nsubj)

    # Likelihood - I think I just need help here.
    y_like = pm.Normal('y_like', mu = b0[subjID] + b1[subjID]*std_x, 
        tau = tau[subjID], observed = std_y)

    # Inference
    step = pm.Metropolis()
    trace = pm.sample(sample, step = step)

どんな助けでも大歓迎です。

4

0 に答える 0