Sha256: 851d1cb2b60a5533b907c1421e8aea61c891a365b83251962447c1ae64a0bc48
Contents?: true
Size: 1.19 KB
Versions: 5
Compression:
Stored size: 1.19 KB
Contents
#ifndef LDA_INFERENCE_H #define LDA_INFERENCE_H #include <stdlib.h> #include <math.h> #include <float.h> #include <assert.h> #include "lda.h" #include "utils.h" int LAG = 5; float EM_CONVERGED; int EM_MAX_ITER; int ESTIMATE_ALPHA; double INITIAL_ALPHA; int NTOPICS; float VAR_CONVERGED; int VAR_MAX_ITER; #ifdef USE_RUBY corpus *last_corpus; lda_model *last_model; double **last_gamma; double **last_phi; enum BOOL { FALSE, TRUE } corpus_loaded, model_loaded, VERBOSE; #endif double lda_inference(document*, lda_model*, double*, double**); double compute_likelihood(document*, lda_model*, double**, double*); double doc_e_step(document* doc, double* gamma, double** phi, lda_model* model, lda_suffstats* ss); void save_gamma(char* filename, double** gamma, int num_docs, int num_topics); void run_em(char* start, char* directory, corpus* corpus); #ifdef USE_RUBY void run_quiet_em(char* start, corpus* corpus); #endif void read_settings(char* filename); void infer(char* model_root, char* save, corpus* corpus); #endif
Version data entries
5 entries across 5 versions & 1 rubygems