Exemplos de MellowmaxDistribution em Python

Exemplo n.º 1

Arquivo: test_distribution.py Projeto: ypxie/chainerrl

 def setUp(self):
     self.values = np.random.rand(self.batch_size, self.n)
     if self.wrap_by_variable:
         self.distrib = distribution.MellowmaxDistribution(
             chainer.Variable(self.values))
     else:
         self.distrib = distribution.MellowmaxDistribution(self.values)

Exemplo n.º 2

Arquivo: mellowmax_policy.py Projeto: takuma-ynd/chainerrl

 def __call__(self, x):
     h = self.model(x)
     return distribution.MellowmaxDistribution(h, omega=self.omega)

Exemplo n.º 3

Arquivo: mellowmax_policy.py Projeto: zhexiaozhe/chainerrl

 def __call__(self, x, test=False):
     h = self.model(x, test=test)
     return distribution.MellowmaxDistribution(h, omega=self.omega)