70722a84e7ae18a6faf616bc8f5b802689a67ff8,plots.py,,plot_partial_dependence,#Any#Any#Any#,726

Before Change

    print("Custom 3d plot via ``partial_dependence``")
    print

    plt.style.use("classic")
    fig = plt.figure()

    target_feature = (1, 5)
    pdp, (x_axis, y_axis) = partial_dependence(clf, target_feature,

After Change

    target_feature = (1, 5)
    pdp, axes = partial_dependence(clf, target_feature,
                                   X=X_train, grid_resolution=50)
    XX, YY = np.meshgrid(axes[0], axes[1])
    Z = pdp[0].reshape(list(map(np.size, axes))).T
    ax = Axes3D(fig)
    surf = ax.plot_surface(XX, YY, Z, rstride=1, cstride=1, cmap=plt.cm.BuPu)

Instances