add naive bayes and rnn route for personalized training

TailUFPB · May 25, 2024 · 5dfd7e6 · 5dfd7e6
1 parent 2a831dc
commit 5dfd7e6
Showing 1 changed file with 0 additions and 5 deletions.
diff --git a/api/Neural_Network2.py b/api/Neural_Network2.py
@@ -312,12 +312,7 @@ def validate_epoch(model, valid_loader, criterion):
 
  return total_loss / total
 
-<<<<<<< HEAD
 def create_and_train_rnn_model(df, name, epochs = 10, batch_size = 32, learning_rate = 0.001):
-=======
->>>>>>> dee0a494328004861698578544010c16b2a995da
-
-def create_and_train_model(df, name, epochs=10, batch_size=32, learning_rate=0.001):
  # Configurações iniciais e preparações do modelo
  dropout_probability = 0.2
  n_rnn_layers = 1