def _log_prob(self, counts):
counts = self._maybe_assert_valid_sample(counts)
ordered_prob = (
tf.lbeta(self.concentration + counts)
- tf.lbeta(self.concentration))
return ordered_prob + distribution_util.log_combinations(
self.total_count, counts)
def _log_normalization(self, counts):
counts = self._maybe_assert_valid_sample(counts)
return -distribution_util.log_combinations(self.total_count, counts)
def _log_normalization(self, counts): counts = self._maybe_assert_valid_sample(counts) return -distribution_util.log_combinations(self.total_count, counts)
