summaryrefslogtreecommitdiff
path: root/gi/clda
diff options
context:
space:
mode:
Diffstat (limited to 'gi/clda')
-rw-r--r--gi/clda/src/ccrp.h4
-rw-r--r--gi/clda/src/crp_test.cc1
2 files changed, 5 insertions, 0 deletions
diff --git a/gi/clda/src/ccrp.h b/gi/clda/src/ccrp.h
index 9b1c5284..eeccce1a 100644
--- a/gi/clda/src/ccrp.h
+++ b/gi/clda/src/ccrp.h
@@ -30,6 +30,10 @@ class CCRP {
return it->second.table_counts_.size();
}
+ unsigned num_customers() const {
+ return num_customers_;
+ }
+
// returns +1 or 0 indicating whether a new table was opened
int increment(const Dish& dish, const double& p0, MT19937* rng) {
DishLocations& loc = dish_locs_[dish];
diff --git a/gi/clda/src/crp_test.cc b/gi/clda/src/crp_test.cc
index 750d80a7..ed384f81 100644
--- a/gi/clda/src/crp_test.cc
+++ b/gi/clda/src/crp_test.cc
@@ -79,6 +79,7 @@ TEST_F(CRPTest, Exchangability) {
++hist[c];
tot += c;
}
+ EXPECT_EQ(cust, crp.num_customers());
cerr << "P(a) = " << (xt / samples) << endl;
cerr << "E[num tables] = " << (tot / samples) << endl;
double error = fabs((tot / samples) - 5.4);