diff options
Diffstat (limited to 'gi/clda')
-rw-r--r-- | gi/clda/src/ccrp.h | 4 | ||||
-rw-r--r-- | gi/clda/src/crp_test.cc | 1 |
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); |