2015/08/26
前回のpythonに引き続き、今回はjavaでDropout + ReLUを実装したコードを紹介します。package名は”DeepLearning”としました。下記の４つのファイルで構成されています。
・Dropout.java
・HiddenLayer.java
・LogisticRegression.java
・utils.java
ReLU以外の活性化関数にも対応できるよう、ラムダ式を用いて”activation”という変数に関数を入れています。
そのため、JDK 1.8 以上のみの対応になります。コードは下記です。
[ Dropout.java ]
[ HiddenLayer.java ]
[ LogisticRegression.java ]
[ utils.java ]
GitHubリポジトリも参考ください。