如何在R中為混合模型編寫此函數？

Question

我想要估計R中的模型。

其中一個部分是有限混合模型，它由兩個OLS組成 。

作為R的新生，我不知道如何在R中寫出這個概率密度函數。

我想知道你是否可以提供一些幫助。

概率密度函數如下：

f(y|x)=(p/σ1)*φ(y-x*b1/σ1)+((1-p)/σ2)*φ(y-x*b2/σ2)

我用stata寫了一個例子：

 
   gen double f1'=normalden($ML_y1,xb1',exp(lns1')) gen doublef2'=normalden($ML_y1,xb2',exp(lns2'))
   tempvar p
   gen double p'=exp(lp')/(1+exp(lp')) replacelnf'=ln(p'*f1'+(1-p')*f2') 
   

我想知道你是否可以告訴我如何在R中編寫這個函數

非常感謝，我期待着您的幫助

Answer 1

請參閱函數FLXMRglm 。 密度用dnorm估算

library(flexmix)
FLXMRglm

# your case 


if (family == "gaussian") {
        z@defineComponent <- function(para) {
            predict <- function(x, ...) {
                dotarg = list(...)
                if ("offset" %in% names(dotarg)) 
                  offset <- dotarg$offset
                p <- x %*% para$coef
                if (!is.null(offset)) 
                  p <- p + offset
                p
            }
            logLik <- function(x, y, ...) dnorm(y, mean = predict(x, 
                ...), sd = para$sigma, log = TRUE)
            new("FLXcomponent", parameters = list(coef = para$coef, 
                sigma = para$sigma), logLik = logLik, predict = predict, 
                df = para$df)
        }
        z@fit <- function(x, y, w, component) {
            fit <- lm.wfit(x, y, w = w, offset = offset)
            z@defineComponent(para = list(coef = coef(fit), df = ncol(x) + 
                1, sigma = sqrt(sum(fit$weights * fit$residuals^2/mean(fit$weights))/(nrow(x) - 
                fit$rank))))
        }
    }