def __init__(self, *args, **kwargs): self.discriminator = None self.generator = None self.loss = None self.trainer = None BaseGAN.__init__(self, *args, **kwargs) self.x, self.y = self.inputs.next()
def test_create_component(self): with self.test_session(): gan = BaseGAN(inputs=MockInput()) distribution = gan.create_component(gan.config.distribution) self.assertEqual( type(distribution), hg.distributions.uniform_distribution.UniformDistribution)
def __init__(self, latent=None, x=None, *args, **kwargs): self.discriminator = None self.latent = None self.generator = None self.loss = None self.trainer = None self.session = None self.latent = latent self.x = x BaseGAN.__init__(self, *args, **kwargs)
def __init__(self, latent=None, x=None, *args, **kwargs): self.discriminator = None self.latent = None self.generator = None self.loss = None self.trainer = None self.session = None self.latent = latent self.x = x BaseGAN.__init__(self, *args, **kwargs)
def test_create(self): with self.test_session(): gan = BaseGAN(inputs=MockInput()) gan.create() self.assertEqual(gan.created, True) with self.assertRaises(ValidationException): gan.create()
def test_get_config_value(self): with self.test_session(): gan = BaseGAN(inputs=MockInput()) self.assertEqual(type(gan.get_config_value('generator')), hc.config.Config) self.assertEqual(gan.get_config_value('missing'), None)
def test_height(self): with self.test_session(): gan = BaseGAN(inputs=MockInput()) self.assertEqual(gan.height(), 32)
def test_batch_size(self): with self.test_session(): gan = BaseGAN(inputs=MockInput(batch_size=1)) self.assertEqual(gan.batch_size(), 1)
def test_has_input(self): with self.test_session(): inputs = MockInput() gan = BaseGAN(inputs=inputs) self.assertEqual(gan.inputs.x, inputs.x)
def test_height(self): with self.test_session(): gan = BaseGAN(inputs = MockInput()) self.assertEqual(gan.height(), 32)
def __init__(self, *args, **kwargs): self.frames = kwargs.pop('frames') BaseGAN.__init__(self, *args, **kwargs)
def test_create_component(self): with self.test_session(): gan = BaseGAN(inputs=MockInput()) encoder = gan.create_component(gan.config.encoder) self.assertEqual(type(encoder), hg.encoders.uniform_encoder.UniformEncoder)
def __init__(self, *args, **kwargs): self.per_sample_frames = 3 self.frames = kwargs.pop('frames') self.per_sample_frames = kwargs.pop('per_sample_frames') BaseGAN.__init__(self, *args, **kwargs)
def __init__(self, *args, **kwargs): BaseGAN.__init__(self, *args, **kwargs) self.x = self.inputs.next()
def test_get_config_value(self): with self.test_session(): gan = BaseGAN(inputs = MockInput()) self.assertEqual(type(gan.get_config_value('generator')), hc.config.Config) self.assertEqual(gan.get_config_value('missing'), None)
def test_constructor(self): with self.test_session(): gan = BaseGAN(inputs=MockInput()) self.assertNotEqual(gan.config.description, None)
def __init__(self, *args, **kwargs): BaseGAN.__init__(self, *args, **kwargs)
def test_batch_size(self): with self.test_session(): gan = BaseGAN(inputs = MockInput(batch_size = 1)) self.assertEqual(gan.batch_size(), 1)