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

modid rn

parent 9e422955
No related branches found
No related tags found
No related merge requests found
import numpy as np import numpy as np
import pandas as pd import pandas as pd
from sklearn.model_selection import train_test_split from sklearn.model_selection import train_test_split, GridSearchCV
from sklearn.preprocessing import StandardScaler from sklearn.preprocessing import StandardScaler
from sklearn.metrics import mean_absolute_error, mean_squared_error, r2_score from sklearn.metrics import mean_absolute_error, mean_squared_error, r2_score
import time import time
...@@ -44,7 +44,7 @@ def train_mlp(df): ...@@ -44,7 +44,7 @@ def train_mlp(df):
# 5. Recherche des meilleurs hyperparamètres avec GridSearchCV # 5. Recherche des meilleurs hyperparamètres avec GridSearchCV
mlp = MLPRegressor(random_state=42) mlp = MLPRegressor(random_state=42)
grid_search = GridSearchCV(mlp, param_grid, cv=3, scoring='r2', n_jobs=-1, verbose=2) grid_search = GridSearchCV(mlp, param_grid, cv=3, scoring='r2', verbose=2)
grid_search.fit(X_train_scaled, y_train) grid_search.fit(X_train_scaled, y_train)
# 6. Affichage des meilleurs paramètres # 6. Affichage des meilleurs paramètres
......
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