chainerでレイヤ数を雑に指定できるMLP

Chainer MLPでgoogleして出てくるのはLinearの数がハードコーディングなMLPばっかりだし、ユニット数を雑に指定して雑にMLP作れるやつが欲しかったので書いた。
もっとましな書き方あったら教えてください。

例えばtanhを活性化関数に用いるユニット数が128, 32, 64の二層MLPなら以下のように書ける。

MLPは最後の出力にsoftmaxを取ることが多いので、最後の活性化関数が必要ないなら以下のようにする。

 

コメントを残す

メールアドレスが公開されることはありません。 * が付いている欄は必須項目です