我是 stan 和机器学习的新手。现在我想实现 pmf 模型。这是我的代码的一部分:
pmf_cod="""
data {
int<lower=0> K; //number of factors
int<lower=0> N; //number of user
int<lower=0> M; //number of item
int<lower=0> D; //number of observation
int<lower=0> D_new; //number of pridictor
int<lower=0, upper=N> ii[D]; //item
int<lower=0, upper=M> jj[D]; //user
int<lower=0, upper=N> ii_new[D_new]; // item
int<lower=0, upper=N> jj_new[D_new]; // user
real<lower=0, upper=5> r[D]; //rating
real<lower=0, upper=5> r_new[D_new]; //pridict rating
}
parameters {
row_vector[K] i[M]; // item profile
row_vector[K] u[N]; // user profile
real<lower=0> alpha;
real<lower=0> alpha_i;
real<lower=0> alpha_u;
}
transformed parameters {
matrix[N,M] I; // indicator variable
I <- rep_matrix(0, N, M);
for (d in 1:D){
I[ii[d]][jj[d]] <- 1;
}
}
model {
for (d in 1:D){
r[d] ~ normal(sum(u[jj[d]]' * i[ii[d]]), 1/alpha);
}
for (n in 1: N){
u[n] ~ normal(0,(1/alpha_u) * I);
}
for (m in 1:M){
i[m] ~ normal(0,(1/alpha_i) * I);
}
}
"""
但是我得到了一个错误:No matches for: row vector ~ normal(int, matrix)
在这行代码中:
for (n in 1: N){
u[n] ~ normal(0,(1/alpha_u) * I);
}
哪里I
是单位矩阵。所以 的乘积(1/alpha_u) * I
也是一个矩阵。但是 stan 只是接受向量或实数值作为方差。我想知道如何将其转换为向量或单个值。
先感谢您!