From 289f96779e665ba24adca3461a624c68aa37bd99 Mon Sep 17 00:00:00 2001
From: Chris Dyer <cdyer@cs.cmu.edu>
Date: Sat, 10 Mar 2012 14:10:04 -0500
Subject: do Bayesian inference on quasimodel2 hyperparameters

---
 gi/pf/tied_resampler.h | 11 +++++++++++
 1 file changed, 11 insertions(+)

(limited to 'gi/pf/tied_resampler.h')

diff --git a/gi/pf/tied_resampler.h b/gi/pf/tied_resampler.h
index 5a262f9d..6f45fbce 100644
--- a/gi/pf/tied_resampler.h
+++ b/gi/pf/tied_resampler.h
@@ -42,6 +42,10 @@ struct TiedResampler {
     return llh;
   }
 
+  double LogLikelihood() const {
+    return LogLikelihood(discount, strength);
+  }
+
   struct DiscountResampler {
     DiscountResampler(const TiedResampler& m) : m_(m) {}
     const TiedResampler& m_;
@@ -106,6 +110,13 @@ struct BinTiedResampler {
     }
   }
 
+  double LogLikelihood() const {
+    double llh = 0;
+    for (unsigned i = 0; i < resamplers.size(); ++i)
+      llh += resamplers[i].LogLikelihood();
+    return llh;
+  }
+
  private:
   std::vector<TiedResampler<CRP> > resamplers;
 };
-- 
cgit v1.2.3