add Dens layers

backend/IT18218640/train.py

@@ -12,7 +12,7 @@ LEARNING_RATE = 0.0001
 EPOCHS = 60
 BATCH_SIZE = 32
 
-NUM_KEYWORDS = 10
+NUM_KEYWORDS = 6
 
 def load_dataset(data_path):
 
@@ -67,6 +67,12 @@ def build_model(input_shape, learning_rate, error="sparse_categorical_crossentro
     model.add(keras.layers.Dense(64, activation="relu"))
     model.add(keras.layers.Dropout(0.3))
 
+    # model.add(keras.layers.Flatten())
+    # model.add(keras.layers.Dense(128, activation="relu"))
+    # model.add(keras.layers.Dense(64, activation="relu"))
+    # model.add(keras.layers.Dropout(0.3))
+    # model.add(keras.layers.Dense(16, activation="relu"))
+
     # softmax classifier
     model.add(keras.layers.Dense(NUM_KEYWORDS, activation="softmax"))