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

Version Path
ealdent-lda-ruby-0.1.6 lib/lda-inference.h
ealdent-lda-ruby-0.1.7 lib/lda-inference.h
ealdent-lda-ruby-0.2.0 lib/lda-inference.h
ealdent-lda-ruby-0.2.1 lib/lda-inference.h
ealdent-lda-ruby-0.2.2 lib/lda-inference.h