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)
def __call__(self, x):
h = self.model(x)
return distribution.MellowmaxDistribution(h, omega=self.omega)
def __call__(self, x, test=False):
h = self.model(x, test=test)
return distribution.MellowmaxDistribution(h, omega=self.omega)
