在Stan中编写自定义Probit函数

时间:2019-05-15 21:16:34

标签: stan

我正在尝试在Stan中编写自定义Probit函数,以提高我对Stan语言和可能性的理解。到目前为止,我已经编写了正常pdf的对数,但是在尝试编写可能性时,收到一条错误消息,发现该错误消息无法理解。我在做什么错了?

斯坦模型

functions {
    real normal_lpdf(real mu, real sigma) {
      return -log(2 * pi()) / 2 - log(sigma) 
             - square(mu) / (2 * sigma^2);
    }
    real myprobit_lpdf(int y | real mu, real sigma) {
      return normal_lpdf(mu, sigma)^y * (1 - normal_lpdf(mu, sigma))^(1-y);
    }
}

data {
    int N;
    int y[N];
}

parameters {
    real mu;
    real<lower = 0> sigma;
}

model {
    for (n in 1:N) {
        target += myprobit_lpdf(y[n] | mu, sigma);
    }
}

错误

预期的解析器: stanc中的错误(model_code =粘贴(程序,合拢=“ \ n”),model_name = model_cppname ,: 由于上述错误,无法解析Stan模型“ Probit_lpdf”。

R代码以模拟数据

## DESCRIPTION

# testing a Probit model

## DATA
N <- 2000
sigma <- 1
mu <- 0.3
u <- rnorm(N, 0, 2)
y.star <- rnorm(N, mu, sigma)
y <- ifelse(y.star > 0,1, 0)

data = list(
    N = N,
    y = y
)

## MODEL
out.stan <- stan("Probit_lpdf.stan",data = data, chains = 2, iter = 1000 )

1 个答案:

答案 0 :(得分:2)

完整的错误消息是

SYNTAX ERROR, MESSAGE(S) FROM PARSER:
Probabilty functions with suffixes _lpdf, _lpmf, _lcdf, and _lccdf,
require a vertical bar (|) between the first two arguments.
 error in 'model2a7252aef8cf_probit' at line 7, column 27
  -------------------------------------------------
     5:     }
     6:     real myprobit_lpdf(real y,  real mu, real sigma) {
     7:       return normal_lpdf(mu, sigma)^y * (1 - normal_lpdf(mu, sigma))^(1-y);
                                  ^
     8:     }
  -------------------------------------------------

这告诉您normal_lpdf函数除了三个输入和一个竖线,将第一个与第二个输入分开。

给您的函数起一个与Stan语言中已经存在的函数相同的名称(例如normal_lpdf)也不是一个好主意。

但是您编写的函数始终无法实现概率模型的对数似然性。首先,数据无法识别错误的标准偏差,因此您不需要sigma。然后,正确的表达式将类似于

real Phi_mu = Phi(mu);
real log_Phi_mu = log(Phi_mu);
real log1m_Phi_mu = log1m(Phi_mu);
for (n in 1:N)
  target += y[n] == 1 ? log_Phi_mu : log1m_Phi_mu;

尽管这只是一个很慢的方法

target += bernoulli_lpmf(y | Phi(mu));