INNER CODE UNIT · Python
ind2
XMU-Kuangnan-Fang-Team/GENetLib · GENetLib/BsplineFunc.py:245
ind2 = np.arange(nbasis, nbasis + q)
basismat = np.asmatrix(basismat)
basismat = np.c_[basismat, np.zeros((basismat.shape[0], q))]
basismat[ind1[:, np.newaxis], ind2] = covariates
if matwt:
wtfac = cholesky(wtvec)
basisw = wtvec @ basismat
else:
rtwtvec = np.sqrt(wtvec)
basisw = (wtvec @ np.ones((1, nbasis + q))) * np.array(basismat)
Bmat = basisw.T @ basismat
Bmat0 = Bmat
if len(dimy) < 3:
Dmat = basisw.T @ y
else:
Dmat = np.zeros((nbasis + q, nrep, nvar))
for ivar in range(nvar):
Dmat[:, :, ivar] = basisw.T @ y[:, :, ivar]