Get some posteriors in there as well

Jonathan Chang · Jonathan Chang · commit 14f00dda5d7b · 2011-07-20T16:25:04.000+02:00
diff --git a/DirichletPosterior.cpp b/DirichletPosterior.cpp
@@ -0,0 +1,16 @@
+class DirichletPosterior {
+  Rcpp::Matrix<INTSXP> document_topic_counts_;
+  double alpha_;
+public:
+#error initialize matrix objects
+  DirichletPosterior(const int K, double alpha) : topics(K), alpha_(alpha_) {
+  }
+
+  virtual double evaluate(const int topic, const Position& position) const {
+    return document_topic_counts_(topic, position.getDocument()) + alpha;
+  }
+
+  virtual void update(const int topic, const Position& position, const int count) {
+    document_topic_counts_(topic, position.getDocument()) += count;
+  }
+};
diff --git a/MatrixBackedTopicCounts.cpp b/MatrixBackedTopicCounts.cpp
@@ -11,6 +11,7 @@ class MatrixBackedTopicCounts : public TopicCounts {
   }
 
   virtual void update(const int topic, const int word, const int count) {
-    
+    topic_word_counts_(topic, word) += count;
+    topic_word_counts_(topic) += count;
   }
 };
diff --git a/MultinomialPosterior.cpp b/MultinomialPosterior.cpp
@@ -0,0 +1,10 @@
+class MultinomialPosterior {
+  MatrixBackedTopicCounts topics;
+public:
+  MultinomialPosterior(const int K_) : topics(K_) {
+  }
+
+  virtual double evaluate(const Position& position, const int topic) const {
+    position.getDocument()
+  }
+};
diff --git a/Position.cpp b/Position.cpp
@@ -0,0 +1,4 @@
+class Position {
+public:
+  virtual int getDocument() const = 0;
+};
diff --git a/Posterior.cpp b/Posterior.cpp
@@ -0,0 +1,5 @@
+class Posterior {
+public:
+  virtual double evaluate(const Position& position, const int topic) const = 0;
+  virtual void update(const Position& position, const int topic, const int count) = 0;
+};
diff --git a/TopicCounts.cpp b/TopicCounts.cpp
@@ -1,5 +1,9 @@
 class TopicCounts {
+  int K_;
 public:
+  TopicCounts(const int K) : K_(K) {
+  }
+
   virtual int getTopicWordCount(const int topic, const int word) const = 0;
   virtual int getTopicCount(const int topic) const = 0;
   virtual void update(const int topic, const int word, const int count) = 0;

Original file line number	Diff line number	Diff line change
`@@ -11,6 +11,7 @@ class MatrixBackedTopicCounts : public TopicCounts {`
`11`	`11`	`}`
`12`	`12`
`13`	`13`	`virtual void update(const int topic, const int word, const int count) {`
`14`		`-`
	`14`	`+ topic_word_counts_(topic, word) += count;`
	`15`	`+ topic_word_counts_(topic) += count;`
`15`	`16`	`}`
`16`	`17`	`};`
-Original file line number
+Diff line change
@@ @@ -0,0 +1,4 @@ @@
 +class Position {
 +public:
 +  virtual int getDocument() const = 0;
 +};