R/predict.bbase.psanova.bs.R

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

predict_bbase.psanova.bs <-
function(object, newx1, newx2) {
	
	MM1p <- predict_bbase.bs(object[[1]], newx1)
	X1p <- attr(MM1p, "X")
	Z1p <- attr(MM1p, "Z")
	
	MM2p <- predict_bbase.bs(object[[2]], newx2)
	X2p <- attr(MM2p, "X")
	Z2p <- attr(MM2p, "Z")
	
	one1. <- X1p[,1, drop = FALSE]
	one2. <- X2p[,1, drop = FALSE]
		
	x1. <- X1p[,-1, drop = FALSE]
	x2. <- X2p[,-1, drop = FALSE]

	Xp <- Rten2(X2p, X1p)
	
	# Delete the intercept
	Xp <- Xp[,-1,drop = FALSE]

	Zp <- cbind(Rten2(one2., Z1p), Rten2(Z2p, one1.), Rten2(x2., Z1p), Rten2(Z2p, x1.), Rten2(Z2p, Z1p))

	B <- cbind(Xp, Zp)
	# Center the matrix
	cm <- attr(object, "cm")
	B <- sweep(B, 2, cm)
	B
}