ealdent-lda-ruby 0.1.5 → 0.1.6

data/lib/lda-inference.c

@@ -546,12 +546,14 @@ void run_quiet_em(char* start, corpus* corpus) {
 
 		last_model = model;
 		last_gamma = var_gamma;
+        last_phi = phi;
 	}
 
 	// output the final model
 
 	last_model = model;
 	last_gamma = var_gamma;
+    last_phi = phi;
 
 	// output the word assignments (for visualization)
 	/*
@@ -835,6 +837,27 @@ static VALUE wrap_get_gamma(VALUE self) {
 	return arr;
 }
 
+static VALUE wrap_get_phi(VALUE self) {
+    if (!model_loaded)
+        return Qnil;
+    
+    VALUE arr;
+    int i = 0, j = 0;
+    int max_length = max_corpus_length(last_corpus);
+    
+    
+    arr = rb_ary_new2(max_length);
+    for (i = 0; i < max_length; i++) {
+        VALUE arr2 = rb_ary_new2(last_model->num_topics);
+        for (j = 0; j < last_model->num_topics; j++) {
+            rb_ary_store(arr2, j, rb_float_new(last_phi[i][j]));
+        }
+        rb_ary_store(arr, i, arr2);
+    }
+    
+    return arr;
+}
+
 /*
  * Get the beta matrix after the model has been run.
  */
@@ -940,6 +963,7 @@ void Init_lda_ext() {
 	// retrieve model and gamma
 	rb_define_method(rb_cLda, "beta", wrap_get_model_beta, 0);
 	rb_define_method(rb_cLda, "gamma", wrap_get_gamma, 0);
+    rb_define_method(rb_cLda, "phi", wrap_get_phi, 0);
 	rb_define_method(rb_cLda, "model", wrap_get_model_settings, 0);
 }
 

data/lib/lda-inference.h

@@ -24,6 +24,7 @@ int VAR_MAX_ITER;
 corpus *last_corpus;
 lda_model *last_model;
 double **last_gamma;
+double **last_phi;
 
 enum BOOL { FALSE, TRUE } corpus_loaded, model_loaded, VERBOSE;
 #endif

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: ealdent-lda-ruby
 version: !ruby/object:Gem::Version 
-  version: 0.1.5
+  version: 0.1.6
 platform: ruby
 authors: 
 - Jason M. Adams
@@ -10,7 +10,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2008-11-21 00:00:00 -08:00
+date: 2008-11-24 00:00:00 -08:00
 default_executable: 
 dependencies: []