Update sequence.py fix bugs of keras

shenweichen · Jan 31, 2024 · 6f66590 · 6f66590
1 parent e8f4d81
commit 6f66590
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepctr/layers/sequence.py b/deepctr/layers/sequence.py
@@ -15,7 +15,7 @@
 except ImportError:
  from tensorflow.python.ops.init_ops_v2 import TruncatedNormal, Constant, glorot_uniform
 
-from tensorflow.python.keras.layers import LSTM, Lambda, Layer, Dropout
+from keras.layers import LSTM, Lambda, Layer, Dropout
 
 from .core import LocalActivationUnit
 from .normalization import LayerNormalization