def write_data(data_dir, idx, W, L, T, D, N, phi, alpha, beta, chains):
B = generate_lda(T, W, D, N, phi, alpha)
pickle.dump(B, open(os.path.join(data_dir, 'counts.pkl'), 'w'))
write_pb(data_dir, idx, W, T, D, alpha, beta, B)
write_stan(data_dir, idx, W, T, D, alpha, beta, B, chains=chains)
write_prism(data_dir, idx, W, T, D, alpha, beta, B)
write_txt(data_dir, idx, B, T, alpha, beta)
def write_data(data_dir, idx, W, L, T, D, N, phi, alpha, beta, chains):
B = generate_lda(T, W, D, N, phi, alpha)
write_pb(data_dir, idx, W, T, D, alpha, beta, B, write_params=False)
write_stan(data_dir, idx, W, T, D, alpha, beta, B, chains=chains, write_params=False)
write_prism(data_dir, idx, W, T, D, alpha, beta, B, write_params=False)
write_txt(data_dir, idx, B, T, alpha, beta, write_params=False)