Merge pull request tensorflow#4126 from npapernot/master

ilyamironov · web-flow · commit fea6447a903e · 2018-04-29T15:35:33.000-07:00
fix import errors and type error in numpy for the PATE implementation
diff --git a/research/differential_privacy/__init__.py b/research/differential_privacy/__init__.py
@@ -0,0 +1 @@
+
diff --git a/research/differential_privacy/multiple_teachers/__init__.py b/research/differential_privacy/multiple_teachers/__init__.py
@@ -0,0 +1 @@
+
diff --git a/research/differential_privacy/multiple_teachers/analysis.py b/research/differential_privacy/multiple_teachers/analysis.py
@@ -240,7 +240,7 @@ def main(unused_argv):
     counts_mat = np.zeros((n, 10)).astype(np.int32)
     for i in range(n):
       for j in range(num_teachers):
-        counts_mat[i, input_mat[j, i]] += 1
+        counts_mat[i, int(input_mat[j, i])] += 1
   n = counts_mat.shape[0]
   num_examples = min(n, FLAGS.max_examples)
 
