def test_softmax_layer_name_is_softmax(self):
     model = KerasModelWrapper(self.model)
     softmax_name = model._get_softmax_name()
     self.assertEqual(softmax_name, 'softmax')
 def test_softmax_layer_name_is_softmax(self):
     model = KerasModelWrapper(self.model)
     softmax_name = model._get_softmax_name()
     self.assertEqual(softmax_name, 'softmax')