if FLAGS.dataset == 'mnist': # ################# # # MNIST Dataset # # ################# # trX, trY, vlX, vlY, teX, teY = datasets.load_mnist_dataset( mode='supervised') elif FLAGS.dataset == 'cifar10': # ################### # # Cifar10 Dataset # # ################### # trX, trY, teX, teY = datasets.load_cifar10_dataset(FLAGS.cifar_dir, mode='supervised') vlX = teX[:5000] # Validation set is the first half of the test set vlY = teY[:5000] elif FLAGS.dataset == 'custom': # ################## # # Custom Dataset # # ################## # def load_from_np(dataset_path): if dataset_path != '': return np.load(dataset_path) else: return None
if FLAGS.dataset == 'mnist': # ################# # # MNIST Dataset # # ################# # trX, vlX, teX = datasets.load_mnist_dataset(mode='unsupervised') width, height = 28, 28 elif FLAGS.dataset == 'cifar10': # ################### # # Cifar10 Dataset # # ################### # trX, teX = datasets.load_cifar10_dataset(FLAGS.cifar_dir, mode='unsupervised') vlX = teX[:5000] # Validation set is the first half of the test set width, height = 32, 32 elif FLAGS.dataset == 'custom': # ################## # # Custom Dataset # # ################## # def load_from_np(dataset_path): if dataset_path != '': return np.load(dataset_path) else: return None