fixed an error

Author: linxihui

keras/layers/crf.py

@@ -119,9 +119,9 @@ def __init__(self, output_dim,
                  weights=None, input_dim=None, input_length=None, unroll=False, **kwargs):
         self.supports_masking = True
         self.output_dim = output_dim
-        self.learn_mode = learn_mode.lower()
-        self.test_mode = test_mode.lower()
+        self.learn_mode = learn_mode
         assert self.learn_mode in ['join', 'marginal']
+        self.test_mode = test_mode
         if self.test_mode is None:
             self.test_mode = 'viterbi' if self.learn_mode == 'join' else 'marginal'
         else:

tests/keras/layers/test_crf.py

@@ -10,7 +10,6 @@
 
 @keras_test
 def test_CRF():
-
     # data
     x = np.random.randint(1, embedding_num, nb_samples * timesteps).reshape((nb_samples, timesteps))
     x[0, -4:] = 0 # right padding
@@ -44,7 +43,7 @@ def test_CRF():
     # test `viterbi_acc
     _, v_acc, _ = model.evaluate(x, y)
     np_acc = (y_pred[x > 0] == y[x > 0]).mean()
-    assert_allclose([v_acc], [np_acc], atol=1e-6)
+    assert np.abs(v_acc - np_acc) < 1e-4
 
     # test config
     model.get_config()