def _load_dataset(self):
self.train_data_gen, self.dev_gen, _ = get_generators(
'f-mnist', self.batch_size)
self.train_gen_test, self.dev_gen_test, self.test_gen_test = \
get_generators(
'f-mnist', self.test_batch_size,
randomize=False)
def _load_dataset(self):
self.train_data_gen, self.dev_gen, self.test_gen = get_generators(
self.dataset_name, self.batch_size, attribute=self.attribute)
self.train_gen_test, self.dev_gen_test, self.test_gen_test = \
get_generators(
self.dataset_name,
self.test_batch_size,
randomize=False,
attribute=self.attribute)
self.test_decoder_images, _ = self.dev_gen().next()
def _load_dataset(self):
"""Loads the dataset."""
self.train_data_gen, self.dev_gen, _ = get_generators(
self.dataset_name,
self.batch_size,
)
self.train_gen_test, self.dev_gen_test, self.test_gen_test = \
get_generators(
self.dataset_name, self.test_batch_size, randomize=False,
)