Add default loss for RNNOutput

vovietanh · May 2, 2016 · bb723e8 · bb723e8
1 parent 3996c61
commit bb723e8
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/deeplearning4j-core/src/main/java/org/deeplearning4j/nn/conf/layers/RnnOutputLayer.java b/deeplearning4j-core/src/main/java/org/deeplearning4j/nn/conf/layers/RnnOutputLayer.java
@@ -18,7 +18,8 @@ private RnnOutputLayer(Builder builder){
 
 	@NoArgsConstructor
     public static class Builder extends BaseOutputLayer.Builder<Builder> {
-
+        protected LossFunction lossFunction = LossFunction.MCXENT;
+
     	public Builder(LossFunction lossFunction) {
             this.lossFunction = lossFunction;
         }