Revision bc4ae43da0ba264e180a4f05f33f25a7f53724e2 authored by Susanne Zabel on 30 March 2021, 09:06:11 UTC, committed by Susanne Zabel on 30 March 2021, 09:06:11 UTC
1 parent b252134
data_mnist.py
#!/usr/bin/python
# from mnist import MNIST
# import matplotlib.pyplot as plt
# import numpy as np
# from itertools import compress
#
# def get_mnist_dataset(digit, n_samples):
# mndata = MNIST('/share/home/zabel/MXNet/data/MNIST')
# images, labels = mndata.load_training()
# n_labels = 10
# image_size = 28
# image_pixels = image_size * image_size
#
# # get only specified digit
# fives_b = [True if labels[i] == digit else False for i in range(len(labels))]
# fives_img = list(compress(images, fives_b))[0:n_samples]
# fives_lab = list(compress(labels, fives_b))[0:n_samples]
# return image_size, image_pixels, fives_lab, fives_img
Computing file changes ...
