Model files have new parts

model/dnn.py

@@ -1 +1,20 @@
+#-- coding: utf-8 -*-
 
+# This code defines a simple Deep Neural Network (DNN) model using Keras.
+
+import keras
+
+class DNN(keras.Model):
+    def __init__(self, input_shape, layer_stack_size, num_classes):
+        super(DNN, self).__init__()
+        self.input_layer = keras.layers.InputLayer(input_shape=input_shape)
+        for i in range(layer_stack_size):
+            setattr(self, f"hidden_layer{i+1}",
+                    keras.layers.Dense(input_shape, activation="relu"))
+        self.output_layer = keras.layers.Dense(num_classes, activation="softmax")
+
+    def call(self, inputs):
+        x = self.input_layer(inputs)
+        x = self.hidden_layer1(x)
+        x = self.hidden_layer2(x)
+        return self.output_layer(x)

model/model.py

@@ -98,6 +98,7 @@ def _transformerblocks(self, inputs, head_size, num_heads,
         Returns:
             A model layer.
         """
+        inputs = Masking(mask_value=pad_value)(inputs)
         x = MultiHeadAttention(
                 key_dim=head_size, num_heads=num_heads,
                 dropout=dropout)(inputs, inputs)