我在阅读《机器学习导论》这本书，迈出了学习机器学习的第一步。我试图生成书中第10页代码片段的图片，该片段可以在这个页面找到，但它不起作用。我说不起作用的意思是，当我点击“运行”时，没有任何东西显示出来（甚至没有错误消息）。

以下代码有什么问题？

我认为我可能缺少像plt.show()这样的代码，但我在谷歌上查找后发现，mglearn似乎不需要/没有这种结构。

from sklearn.neighbors import KNeighborsClassifierimport matplotlib.pyplot as pltimport mglearn.plotsX, y = mglearn.datasets.make_forge()mglearn.discrete_scatter(X[:, 0], X[:, 1], y)plt.legend(["Class 0", "Class 1"], loc=4)plt.xlabel("First feature")plt.ylabel("Second feature")print("X.shape: {}".format(X.shape))#plt.show()mglearn.plots.plot_knn_classification(n_neighbors=1)

我使用的是Python 3.6.3版本。

回答：

根据@Sascha的建议，我得到了以下可用的代码：

from sklearn.neighbors import KNeighborsClassifierimport matplotlib.pyplot as pltimport mglearn.plotsX, y = mglearn.datasets.make_forge()mglearn.plots.plot_knn_classification(n_neighbors=1)plt.show()