Cannot set ylim for sklearn partial dependence plot

Cannot set ylim for sklearn partial dependence plot

Problem Description:

I am following this example on sklearn documentation

I want to change the limits of y axis so I can visually compare results from different models.

import pandas as pd
import matplotlib.pyplot as plt
from sklearn.datasets import load_diabetes
from sklearn.tree import DecisionTreeRegressor
from sklearn.inspection import PartialDependenceDisplay

diabetes = load_diabetes()
X = pd.DataFrame(, columns=diabetes.feature_names)
y =

tree = DecisionTreeRegressor(), y)

fig, ax = plt.subplots(figsize=(12, 6))
tree_disp = PartialDependenceDisplay.from_estimator(tree, X, ["age", "bmi"], ax=ax)

However, it seems that ax.set_ylim get ignored no matter what I specify. On the other hand, ax.set_title given in example works fine.

Solution – 1

PartialDependenceDisplay have an axes_ attribute that represents both matplotlib’s axes of the figure.

You can modify them as follow:

tree_disp = PartialDependenceDisplay.from_estimator(tree, X, ["age", "bmi"], ax=ax)

This will output the following plot:

enter image description here

Rate this post
We use cookies in order to give you the best possible experience on our website. By continuing to use this site, you agree to our use of cookies.