EXA remove warnings by setting n_quantiles in plot_transformed_target (scikit-learn#14156)

CateGitau · glemaitre · commit 55a7752f2ee8 · 2019-06-24T11:49:14.000+02:00
diff --git a/examples/compose/plot_transformed_target.py b/examples/compose/plot_transformed_target.py
@@ -138,7 +138,9 @@
 X = dataset.data[:, np.logical_not(target)]
 y = dataset.data[:, target].squeeze()
 y_trans = quantile_transform(dataset.data[:, target],
-                             output_distribution='normal').squeeze()
+                             n_quantiles=300,
+                             output_distribution='normal',
+                             copy=True).squeeze()
 
 ###############################################################################
 # A :class:`sklearn.preprocessing.QuantileTransformer` is used such that the
@@ -184,7 +186,8 @@
 
 regr_trans = TransformedTargetRegressor(
     regressor=RidgeCV(),
-    transformer=QuantileTransformer(output_distribution='normal'))
+    transformer=QuantileTransformer(n_quantiles=300,
+                                    output_distribution='normal'))
 regr_trans.fit(X_train, y_train)
 y_pred = regr_trans.predict(X_test)
 
