コード例 #1
0
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)
コード例 #2
0
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)