【Python】ソフトマックス関数を実装するコード【Softmax】

スポンサーリンク

【Python】ソフトマックス関数を実装するコード

予測モデルの作成でソフトマックス回帰を実装するのにシグモイド関数を使いたいときに参考として実装プログラムをまとめておく。

ソフトマックス関数の実装

ソフトマックス関数は以下で表される。

これをPythonでコードを書くと以下のようになる。

def softmax(x):
    x -= x.max(axis=1, keepdims=True) # expのoverflowを防ぐ
    x_exp = np.exp(x)
    return x_exp / np.sum(x_exp, axis=1, keepdims=True)

ここでのxは2次元ベクトルで、x -= x.max(axis=1, keepdims=True)をすることによってexpのoverflowを防いでいる。

expを使っているのでシグモイド関数と同様にoverflowに気を付ける必要がある。

以上がソフトマックス関数を実装するコードの紹介。

人気記事

人気記事はこちら。

CUDA、cuDNNのバージョンをターミナルで調べるコマンド
【Pytorch】テンソルの次元を追加・削除する方法【dim】
【Pytorch】テンソルを連結する方法(cat・stack)
【Python】Tensorflowをダウングレード・アップグレードするコマンド
【Protobuf】"TypeError: Descriptors cannot not be created directly."を解決する【solved】

最新記事

最新記事はこちら。

【イタリア】ピサの斜塔へ。中に入って頂上まで上り景色を見る。
【イタリア】サンタ・マリア・デル・フィオーレ大聖堂にある「最後の審判」、屋上から見る景色が良かった
【イタリア】ミケランジェロ広場から夜景を見る。ヴェッキオ橋も。
【イタリア】フィレンツェのフードコートで夕食。ビール・ワイン・海鮮フライなど
【Latex】コンパイルエラーが出た場合の解決方法。Tex Live Managerでインストール。

タイトルとURLをコピーしました