Skip to content

Commit 0f52b69

Browse files
committed
restore cnn_test
1 parent 253af59 commit 0f52b69

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

test/cnn_test.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -41,11 +41,11 @@ def test_train_set_1(self):
4141
mod = cnn.load_model(name)
4242
result = mod.classify_image("photos/DSC86.jpg")
4343
print("result: " + str(result))
44-
self.assertTrue(result[0][0] == "kiwi" and result[0][1] == 1.0)
44+
self.assertTrue(result[0][0] == "kiwi" and result[0][1] > 0.9)
4545

4646
cnn = cnn_manager.CNNManager.get_instance()
47-
#cnn.delete_model(name)
48-
#self.assertTrue(cnn.get_models().get(name) is None)
47+
cnn.delete_model(name)
48+
self.assertTrue(cnn.get_models().get(name) is None)
4949

5050
"""
5151
def test_train_set_2(self):

0 commit comments

Comments
 (0)