def __init__()

in train/CustomModel.py [0:0]

• 4 lines of code
• 1 McCabe index (conditional complexity)

    def __init__(self, return_sequences=True,activation=None):
        self.return_sequences = return_sequences
        self.activation = tf.keras.activations.get(activation)
        super(attention,self).__init__()