def partial_slda_e_step(global_iterations, v):
local_i = 0
for d, (document,y) in enumerate(v.iterdocs()):
local_i = topiclib.partial_slda_E_step_for_doc(global_iterations,
local_i,
d, document, y,
v.alpha, v.beta,
v.gamma[d], v.phi[d],
v.eta, v.sigma_squared)
return local_i
def tlc_e_step(global_iterations, v):
total_local_i = 0
dlocal_i, clocal_i, local_i, blocal_i = 0,0,0,0
for d, (document, comment) in enumerate(v.iterdocuments()):
# todo: this should be more complicated in order to share strength
dlocal_i = topiclib.lda_E_step_for_doc(global_iterations,
dlocal_i,
d, document,
v.alphaU, v.beta[:v.Ku],
v.gammaD[d], v.phiD[d])
alphaC = np.concatenate((v.alphaU, v.alphaS))
clocal_i = topiclib.lda_E_step_for_doc(global_iterations,
clocal_i,
d, comment,
alphaC, v.beta[:v.Kc],
v.gammaC[d], v.phiC[d])
total_local_i += dlocal_i
total_local_i += clocal_i
for l, (labeled_doc,y) in enumerate(v.iterlabeled()):
alphaL = np.concatenate((v.alphaS, v.alphaB))
local_i = topiclib.partial_slda_E_step_for_doc(global_iterations,
local_i,
l, labeled_doc, y,
alphaL, v.beta[-v.Kl:],
v.gammaL[l], v.phiL[l],
v.eta, v.sigma_squared)
total_local_i += local_i
for b, bg_doc in enumerate(v.iterbackground()):
blocal_i = topiclib.lda_E_step_for_doc(global_iterations,
blocal_i,
b, bg_doc,
v.alphaB, v.beta[-v.Kb:],
v.gammaB[b], v.phiB[b])
total_local_i += blocal_i
return total_local_i
