RNN first draft

TailUFPB · Mar 28, 2024 · f80f5ca · f80f5ca
1 parent 1250da2
commit f80f5ca
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/api/Neural_Network2.py b/api/Neural_Network2.py
@@ -95,7 +95,8 @@ def create_and_train_model(train_texts, train_labels, name, epochs=5, batch_size
 
  # Define a arquitetura do modelo
  model = tf.keras.Sequential([
- tf.keras.layers.Dense(64, activation='relu', input_shape=(num_features,)),
+ tf.keras.layers.Embedding(input_dim=num_features, output_dim=64),
+ tf.keras.layers.SimpleRNN(64),
  tf.keras.layers.Dense(num_classes, activation='softmax')
  ])