626_privacy/tensorflow_privacy
2
0
Fork 1
Code Issues Pull requests 1 Projects Releases Packages Wiki Activity Actions

Update input.py

Browse source
This commit is contained in:
Nicolas Papernot 2019-03-18 07:29:18 -07:00 committed by GitHub
parent 5bac796ee6
commit d139d94f32
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
research/pate_2017/input.py
View file

@ -17,8 +17,10 @@ from __future__ import absolute_import
from __future__ import division from __future__ import division
from __future__ import print_function from __future__ import print_function
import cPickle try:
#import pickle #for python 3 import cPickle as pickle
except:
import pickle
import gzip import gzip
import math import math
import numpy as np import numpy as np
@ -156,8 +158,7 @@ def unpickle_cifar_dic(file):
:return: tuple of (images, labels) :return: tuple of (images, labels)
""" """
fo = open(file, 'rb') fo = open(file, 'rb')
dict = cPickle.load(fo) dict = pickle.load(fo)
#dict = pickle.load(fo) #for python 3
fo.close() fo.close()
return dict['data'], dict['labels'] return dict['data'], dict['labels']