R/predict.design.matrix.DDPstar.R

In DDPstar: Density Regression via Dirichlet Process Mixtures of Normal Structured Additive Regression Models

predict.design.matrix.DDPstar <-
function(object, newdata, ...) {
	if(object$fit$mm$iformula$npartial == 0) { # Only the intercept
        Xp <- matrix(1, ncol = 1, nrow = nrow(newdata))
    } else {
		select <- 1:object$fit$mm$iformula$npartial
		Xp <- predict_design.matrix.DDPstar.aux(object = object$fit$mm, newdata = newdata, select = select)
	
		# Add the intercept
		Xp <- cbind(1, Xp)
	}
	res <- list()
	res$X <- Xp
	res
}