Skip to content
Snippets Groups Projects
Commit ce1252e5 authored by Mohamed Sebabti's avatar Mohamed Sebabti
Browse files

model rn

parent d7df2bcb
Branches
No related tags found
No related merge requests found
...@@ -32,7 +32,7 @@ def train_mlp(df): ...@@ -32,7 +32,7 @@ def train_mlp(df):
# 4️⃣ Définition des hyperparamètres # 4️⃣ Définition des hyperparamètres
param_grid = { param_grid = {
"hidden_layer_sizes": [(50,), (100,), (100, 50), (100, 100)], "hidden_layer_sizes": [(50,), (100,), (100, 50), (100, 100)],
"activation": ["relu", "tanh"], "activation": [ "tanh"],
"solver": ["adam", "sgd"], "solver": ["adam", "sgd"],
"learning_rate_init": [0.001, 0.01, 0.1], "learning_rate_init": [0.001, 0.01, 0.1],
"max_iter": [500], "max_iter": [500],
...@@ -65,7 +65,7 @@ def train_mlp(df): ...@@ -65,7 +65,7 @@ def train_mlp(df):
# ⏳ Temps d'exécution # ⏳ Temps d'exécution
elapsed_time = time.time() - start_time elapsed_time = time.time() - start_time
print(f"\n⏱️ Temps d'exécution : {elapsed_time:.2f} secondes") print(f"\n⏱️ Temps d'exécution : {elapsed_time:.2f} secondes")
print(df.head(40))
return df return df
# 📂 Chargement et entraînement # 📂 Chargement et entraînement
......
This diff is collapsed.
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment