lib/dnn/core/activations.rb in ruby-dnn-0.4.4 vs lib/dnn/core/activations.rb in ruby-dnn-0.5.0

@@ -73,10 +73,10 @@
         {name: self.class.name, alpha: alpha}
       end
     end
 
 
-    class IdentityWithLoss < OutputLayer
+    class IdentityMSE < OutputLayer
       def forward(x)
         @out = x
       end
     
       def backward(y)