Sha256: 1005365cab18dbe38352495dd80ca54d875c587e3e482e8a278fd8aec833ffea

Contents?: true

Size: 1.17 KB

Versions: 10

Compression:

Stored size: 1.17 KB

Contents

#ifndef LDA_INFERENCE_H
#define LDA_INFERENCE_H

#include <stdlib.h>
#include <math.h>
#include <float.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**, short*);
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

10 entries across 10 versions & 2 rubygems

Version Path
lda-ruby-0.3.9 ext/lda-ruby/lda-inference.h
ealdent-lda-ruby-0.2.3 lib/lda-inference.h
ealdent-lda-ruby-0.3.0 ext/lda-ruby/lda-inference.h
ealdent-lda-ruby-0.3.1 ext/lda-ruby/lda-inference.h
lda-ruby-0.3.8 ext/lda-ruby/lda-inference.h
lda-ruby-0.3.7 ext/lda-ruby/lda-inference.h
lda-ruby-0.3.6 ext/lda-ruby/lda-inference.h
lda-ruby-0.3.5 ext/lda-ruby/lda-inference.h
lda-ruby-0.3.4 ext/lda-ruby/lda-inference.h
lda-ruby-0.3.1 ext/lda-ruby/lda-inference.h