FIX HistGradientBoosting fitting time too long (INRIA#860)

Author: SebastienMelo

python_scripts/ensemble_hist_gradient_boosting.py

@@ -58,7 +58,7 @@
     data,
     target,
     scoring="neg_mean_absolute_error",
-    n_jobs=2,
+    # n_jobs=2,  # Uncomment this line if you run locally
 )
 
 # %%
@@ -122,7 +122,7 @@
     data,
     target,
     scoring="neg_mean_absolute_error",
-    n_jobs=2,
+    # n_jobs=2,  # Uncomment this line if you run locally
 )
 
 # %%
@@ -161,7 +161,7 @@
     data,
     target,
     scoring="neg_mean_absolute_error",
-    n_jobs=2,
+    # n_jobs=2,  # Uncomment this line if you run locally
 )
 
 # %%

python_scripts/ensemble_hyperparameters.py

@@ -96,7 +96,7 @@
     scoring="neg_mean_absolute_error",
     n_iter=10,
     random_state=0,
-    n_jobs=2,
+    # n_jobs=2,  # Uncomment this line if you run locally
 )
 search_cv.fit(data_train, target_train)
 
@@ -184,7 +184,7 @@
     scoring="neg_mean_absolute_error",
     n_iter=20,
     random_state=0,
-    n_jobs=2,
+    # n_jobs=2, # Uncomment this line if you run locally
 )
 search_cv.fit(data_train, target_train)
 

python_scripts/ensemble_sol_04.py

@@ -70,7 +70,12 @@
 
 cv = KFold(n_splits=5, shuffle=True, random_state=0)
 results = cross_validate(
-    search, data, target, cv=cv, return_estimator=True, n_jobs=2
+    search,
+    data,
+    target,
+    cv=cv,
+    return_estimator=True,
+    # n_jobs=2  # Uncomment this if you run locally
 )
 
 # %% [markdown]