def read_data_sets(data_dir, one_hot, reshape): train_images_file, train_labels_file, test_images_file, test_labels_file = load_mnist_data(data_dir) train_dataset = load_dataset(train_images_file, train_labels_file) train_dataset = train_dataset.shuffle(60000) test_dataset = load_dataset(test_images_file, test_labels_file) mnist = Mnist(train_dataset, test_dataset, one_hot, reshape) return mnist