Skip to content

Commit f2fb60f

Browse files
Irina NicolaeIrina Nicolae
authored andcommitted
Add imports for python2 compatibility of Clever metric (closes #71)
1 parent 2909cb8 commit f2fb60f

File tree

3 files changed

+10
-8
lines changed

3 files changed

+10
-8
lines changed

src/metrics.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22
Module implementing varying metrics for assessing model robustness. These fall mainly under two categories:
33
attack-dependent and attack-independent.
44
"""
5+
from __future__ import absolute_import, division, print_function, unicode_literals
56
import config
67

78
import numpy as np

src/metrics_unittest.py

Lines changed: 8 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,8 @@
1-
# from config import config_dict
1+
from __future__ import absolute_import, division, print_function, unicode_literals
2+
23
import unittest
34

4-
import keras.backend as K
5+
import keras.backend as k
56
import tensorflow as tf
67
import numpy as np
78

@@ -36,7 +37,7 @@ class TestMinimalPerturbation(unittest.TestCase):
3637

3738
def test_emp_robustness_mnist(self):
3839
session = tf.Session()
39-
K.set_session(session)
40+
k.set_session(session)
4041

4142
comp_params = {"loss": 'categorical_crossentropy',
4243
"optimizer": 'adam',
@@ -108,7 +109,7 @@ def test_clever_t_unit(self):
108109
print("Unit test for the targeted version with simplified data.")
109110
# Define session & params
110111
session = tf.Session()
111-
K.set_session(session)
112+
k.set_session(session)
112113

113114
# Get classifier
114115
classifier = TestClassifier()
@@ -129,7 +130,7 @@ def test_clever_u_unit(self):
129130
print("Unit test for the untargeted version with simplified data.")
130131
# Define session & params
131132
session = tf.Session()
132-
K.set_session(session)
133+
k.set_session(session)
133134

134135
# Get classifier
135136
classifier = TestClassifier()
@@ -150,7 +151,7 @@ def test_clever_t(self):
150151
print("Test if the targeted version works on a true classifier/data")
151152
# Define session & params
152153
session = tf.Session()
153-
K.set_session(session)
154+
k.set_session(session)
154155

155156
comp_params = {"loss": 'categorical_crossentropy', "optimizer": 'adam',
156157
"metrics": ['accuracy']}
@@ -178,7 +179,7 @@ def test_clever_u(self):
178179
print("Test if the untargeted version works on a true classifier/data")
179180
# Define session & params
180181
session = tf.Session()
181-
K.set_session(session)
182+
k.set_session(session)
182183

183184
comp_params = {"loss": 'categorical_crossentropy', "optimizer": 'adam',
184185
"metrics": ['accuracy']}

src/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
"""
22
Module providing convenience functions.
33
"""
4-
from __future__ import absolute_import, division, print_function
4+
from __future__ import absolute_import, division, print_function, unicode_literals
55

66
import argparse
77
import json

0 commit comments

Comments
 (0)