INNER CODE UNIT · Python
tnames
XMU-Kuangnan-Fang-Team/GENetLib · GENetLib/BsplineFunc.py:238
tnames = np.arange(1, n + 1)
basismat = eval_basis(argvals, basisobj, 0, returnMatrix)
# Calculation
if n > nbasis + q or lambda_ > 0:
if covariates is not None:
ind1 = np.arange(n)
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