[var dtc missing] yak shaving

This commit is contained in:
mzwiessele 2014-06-27 16:22:27 -07:00
parent 35b778b45f
commit 3ac8f477fa

View file

@ -239,7 +239,7 @@ class VarDTCMissingData(LatentFunctionInference):
logger.info('preparing subarrays Y')
Ys = [Y[v, :][:, ind] for v, ind in self._subarray_indices]
logger.info('preparing traces Y')
traces = [(y**2).sum() for y in Ys]
traces = [np.einsum('ij,ij->', y,y) for y in Ys]
return Ys, traces
else:
self._subarray_indices = [[slice(None),slice(None)]]