I have built an AutoML model, and I would like for it to add Partial Dependency Plots for the final/best model. I am trying to add this to the code generated, but not having any luck. This is the code I attempted.

from sklearn.inspection import PartialDependenceDisplay

# Create Partial Dependency Plots for each class

n_classes = len(model.classes_)

for target_class in range(n_classes):

  target_label = model.classes_[target_class]

  print(f"Partial dependency plots for class {target_label}:")

  display = PartialDependenceDisplay.from_estimator(

    model, X_train, np.arange(X_train.shape[1]), target=target_label

  )

  display.figure_.set_size_inches(12, 😎

  display.figure_.suptitle(f"Partial dependency plots for class {target_label}")

  plt.subplots_adjust(hspace=0.4, wspace=0.4)

  plt.show()