scikit-learn
diff --git a/‎examples/inspection/plot_partial_dependence.py
Copy file name to clipboardExpand all lines: examples/inspection/plot_partial_dependence.py
+7-1Lines changed: 7 additions & 1 deletion b/‎examples/inspection/plot_partial_dependence.py
Copy file name to clipboardExpand all lines: examples/inspection/plot_partial_dependence.py
+7-1Lines changed: 7 additions & 1 deletion
@@ -46,6 +46,10 @@
 # Make an explicit copy to avoid "SettingWithCopyWarning" from pandas
 X, y = bikes.data.copy(), bikes.target
 
+# We use only a subset of the data to speed up the example.
+X = X.iloc[::5, :]
+y = y[::5]
+
 # %%
 # The feature `"weather"` has a particularity: the category `"heavy_rain"` is a rare
 # category.
@@ -286,7 +290,9 @@
 hgbdt_model = make_pipeline(
     hgbdt_preprocessor,
     HistGradientBoostingRegressor(
-        categorical_features=categorical_features, random_state=0
+        categorical_features=categorical_features,
+        random_state=0,
+        max_iter=50,
     ),
 )
 hgbdt_model.fit(X_train, y_train)