コード例 #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)
コード例 #3
0
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)
コード例 #4
0
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)