Make NN model

Details

1. プログラムは、 PyTorchでシンプルな多層ニューラルネットワークを作ろうをベースに作っています。


2. フレームワークは、PyTorch0.4.0を使用しています。


3. データはirisを使用し、4つの特徴（ガク長、ガク幅、花弁長、花弁幅）からアヤメ3種（serosa, verscolor, virginica）を予測します。


4. 引数は以下の通りです。
   • epoch : 学習させる回数 ( > 0 )
   • learning rate : 学習率 ( 0 < lr < 1 )
   • LayerN neurons : 第N隠れ層のノード数 ( > 0 )
   • LayerN's ReLU : チェックでReLU関数を使用 ( on/off )


5. Google colobを使うことで簡単に試せます。
   使い方は、 こちらをご覧ください。