書籍「最短コースでわかる PyTorch &深層学習プログラミング」3値分類モデルの決定境界の表示 - Qiita

はじめに 書籍「最短コースでわかる PyTorch &深層学習プログラミング」の著者です。 この本の姉妹版にあたる「最短コースでわかるディープラーニングの数学」の補足として、３値分類モデルの決定境界を求めるコードをアップしたところ、結構評判がいいようなので、調子にのってこちらの本でも同じことをやってみます。 書籍のリンクはこちら 2つの書籍は、まったく同じデータを例題で扱っており、こちらの本で３値分類モデルは７章に該当します。 Notebook全体のリンクは、こちらです。 考え方 決定境界とは、２つの確率値が等しくなる点の集まりです。つまり、 softmax関数の入力の時点で考えると、「２つの１次関数の計算結果が等しい場所」ということになります。 これを、重み行列とバイアスの言葉でいいかえると、重み行列とバイアスを当時に一列ずらして元の行列との差をとり、その結果で計算される１次関数の結果が