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)
def write_pb_txt(data_dir, idx, W, L, T, D, N, phi, alpha, beta): #B = generate_lda(T, W, D, N, phi, alpha) B,phi = gen_lda2(T, W, D, N, alpha) write_pb(data_dir, idx, W, T, D, alpha, beta, B) write_txt(data_dir, idx, B)
def write_pb_church(data_dir, idx, W, L, T, D, N, phi, alpha, beta, n_samples, lag): B = generate_lda(T, W, D, N, phi, alpha) write_pb(data_dir, idx, W, T, D, alpha, beta, B) write_church(data_dir, idx, B, alpha,beta,D,T,W,N, n_samples, lag) write_church2(data_dir, idx, B, alpha,beta,D,T,W, n_samples, lag)