nikethanreddy commited on
Commit
5e5afe8
·
verified ·
1 Parent(s): 1e1a4ef

Update tkat.py

Browse files
Files changed (1) hide show
  1. tkat.py +1 -1
tkat.py CHANGED
@@ -1,7 +1,7 @@
1
  import tensorflow as tf
2
  from tensorflow.keras.layers import Layer, Dense
3
 
4
- class AttentionLayer(Layer):
5
  def __init__(self, units):
6
  super(AttentionLayer, self).__init__()
7
  self.W1 = Dense(units)
 
1
  import tensorflow as tf
2
  from tensorflow.keras.layers import Layer, Dense
3
 
4
+ class TKAT(Layer):
5
  def __init__(self, units):
6
  super(AttentionLayer, self).__init__()
7
  self.W1 = Dense(units)