質問: 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)) #X.shape: (26, 2) ```
回答: `legend`は凡例を表示します。`loc`は凡例の表示位置を指定します。`loc=4` は `lower right` の意味です。 詳細はmatplotlib.pyplot.legendを参照ください。 `mglearn`とはPythonではじめる機械学習で提供されているライブラリintroduction_to_ml_with_python/mglearn/のことでしょうか? 第二引数までが、特徴量(x,y軸)、第三引数がクラス分類値(0 or 1)のようです。 参考: 機械学習入門_第2章 途中まで introduction_to_ml_with_python/02-supervised-learning.ipynb