Wrong code in [7] to plot the Rosenbrock function ?

[jason990420]

For statements in [7] of https://github.com/SMTorg/smt/blob/master/tutorial/SMT_Tutorial.ipynb, the shapes for x, y are different from z, and it will get following figure which is different from the figure shown on that web page.

ax.scatter(xt[:, 0], xt[:, 1], yt, zdir='z', marker='x', c='b', s=200, label='Training point')
ax.scatter(xtest[:, 0], xtest[:, 1], ytest, zdir='z', marker='.', c='k', s=200, label='Validation point')

Maybe it should be revised as following, but it looks like that there's still a little different.

ax.scatter(xt[:, 0], xt[:, 1], yt[:, 0], zdir='z', marker='x', c='b', s=200, label='Training point')
ax.scatter(xtest[:, 0], xtest[:, 1], ytest[:, 0], zdir='z', marker='.', c='k', s=200, label='Validation point')

[Paul-Saves]

Thank you very much for noticing the problem and proposing a fix! It most likely is related to the release of numpy 2.0.0.

[jason990420]

Installed numpy on my windows: 1.26.4.