Beispiel #1
0
    def test_sample(self):
        def _distribution(param):
            return Multinomial(param, n_experiments=10)

        utils.test_1parameter_sample_shape_one_rank_less(
            self, _distribution, np.zeros)
        dist = Multinomial(np.ones([2, 2]), n_experiments=None)
        with self.assertRaisesRegexp(ValueError,
                                     "Cannot sample when `n_experiments`"):
            dist.sample()
Beispiel #2
0
    def test_sample_shape(self):
        def _distribution(param):
            return Multinomial(param, 10)

        utils.test_1parameter_sample_shape_one_rank_less(
            self, _distribution, np.zeros)
Beispiel #3
0
    def test_sample_shape(self):
        def _proxy_distribution(logits):
            return Concrete(1., logits)

        utils.test_1parameter_sample_shape_one_rank_less(
            self, _proxy_distribution, np.zeros)
Beispiel #4
0
 def test_sample_shape(self):
     utils.test_1parameter_sample_shape_one_rank_less(
         self, Dirichlet, np.zeros)
Beispiel #5
0
 def test_sample_shape(self):
     utils.test_1parameter_sample_shape_one_rank_less(
         self, OnehotCategorical, np.zeros)