ruby/tensorflow_privacy
1
0
Fork 0
forked from 626_privacy/tensorflow_privacy
Code Pull requests Activity

Closes #32

Browse source 
PiperOrigin-RevId: 239054994
This commit is contained in:
Nicolas Papernot 2019-03-18 13:57:47 -07:00 committed by A. Unique TensorFlower
parent f58891f3e3
commit a1e9ca2a18
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
research/pate_2017/input.py
View file

@ -250,7 +250,7 @@ def extract_mnist_data(filename, num_images, image_size, pixel_depth):
np.save(filename, data) np.save(filename, data)
return data return data
else: else:
with tf.gfile.Open(filename+'.npy', mode='r') as file_obj: with tf.gfile.Open(filename+'.npy', mode='rb') as file_obj:
return np.load(file_obj) return np.load(file_obj)
@ -266,7 +266,7 @@ def extract_mnist_labels(filename, num_images):
np.save(filename, labels) np.save(filename, labels)
return labels return labels
else: else:
with tf.gfile.Open(filename+'.npy', mode='r') as file_obj: with tf.gfile.Open(filename+'.npy', mode='rb') as file_obj:
return np.load(file_obj) return np.load(file_obj)