From 5dfd7e61e6bc49b71c7426ba6e72850b7cda4caa Mon Sep 17 00:00:00 2001 From: Gisele Silva Date: Sat, 25 May 2024 03:40:11 -0300 Subject: [PATCH] add naive bayes and rnn route for personalized training --- api/Neural_Network2.py | 5 ----- 1 file changed, 5 deletions(-) diff --git a/api/Neural_Network2.py b/api/Neural_Network2.py index a43c9652..80440fe8 100644 --- 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