如何使用训练/测试数据评估pymc2模型?

时间:2019-01-13 13:11:24

标签: python pymc

我正在pymc2中构建一个简单的模型,我想评估火车数据和测试数据。

我尝试使用这部分代码

print('Accuracy on train data = {}%'.format((y.value == Y_train).mean() * 100))

但是我认为y.valueY_train相同,因此无法解决我的问题。

我当前的代码是

number_of_samples = 10000
X = np.random.randn(100, 2)
Y = np.tanh(X[:, 0] + X[:, 1])
Y = 1. / (1. + np.exp(-(Y + Y)))
Y_train = Y > 0.5

w11 = pm.Normal('w11', mu=0., tau=1.)
w12 = pm.Normal('w12', mu=0., tau=1.)
w21 = pm.Normal('w21', mu=0., tau=1.)
w22 = pm.Normal('w22', mu=0., tau=1.)
w31 = pm.Normal('w31', mu=0., tau=1.)
w32 = pm.Normal('w32', mu=0., tau=1.)

x1 = X[:, 0]
x2 = X[:, 1]

x3 = pm.Lambda('x3', lambda w1=w11, w2=w12: np.tanh(w1 * x1 + w2 * x2))
x4 = pm.Lambda('x4', lambda w1=w21, w2=w22: np.tanh(w1 * x1 + w2 * x2))

@pm.deterministic
def sigmoid(x=w31 * x3 + w32 * x4):
    return 1. / (1. + np.exp(-x))

y = pm.Bernoulli('y', sigmoid, observed=True, value=Y_train)

model = pm.Model([w11, w12, w21, w22, w31, w32, y])
inference = pm.MCMC(model)

inference.sample(number_of_samples)
print('Accuracy on train data = {}%'.format((y.value == Y_train).mean() * 100))

这是我要建立的网络。

my network

我希望根据训练数据和其他测试数据来计算训练后的模型的准确性,但是对我而言我还不清楚如何做到这一点。

1 个答案:

答案 0 :(得分:1)

我认为您可能想要的是posterior predictive check,您可以通过在模型中添加其他随机变量来实现:

y_pred = pm.Bernoulli('y_pred', sigmoid)

model = pm.Model([w11, w12, w21, w22, w31, w32, y, y_pred])

要获得样本内预测,可以使用一些阈值(例如0.5)将y_pred迹线中的概率预测映射到适合测量精度的确定性预测:

y_pred_samples = y_pred.trace()
y_pred_threshold = (y_pred_samples.mean(axis=0) > .5)
print('Accuracy on train data = {}%'.format((y_pred_threshold == Y_train).mean() * 100))

这是一个Jupyter笔记本,将所有内容组合在一起:link