lib/lda-inference.h in ealdent-lda-ruby-0.2.2 vs lib/lda-inference.h in ealdent-lda-ruby-0.2.3

@@ -2,11 +2,10 @@
 #define LDA_INFERENCE_H
 
 #include <stdlib.h>
 #include <math.h>
 #include <float.h>
-#include <assert.h>
 #include "lda.h"
 #include "utils.h"
 
 
 
@@ -29,11 +28,11 @@
 enum BOOL { FALSE, TRUE } corpus_loaded, model_loaded, VERBOSE;
 #endif
 
 
 
-double lda_inference(document*, lda_model*, double*, double**);
+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,
@@ -59,6 +58,6 @@
 void infer(char* model_root,
            char* save,
            corpus* corpus);
            
 
-#endif
\ No newline at end of file
+#endif