Class: RubyZero::NN::Losses::MSE

Inherits:
Loss
  • Object
show all
Defined in:
lib/rubyzero/nn/losses/mse.rb

Instance Method Summary collapse

Methods inherited from Model

#__get_str__, #call, #eval, #initialize, #inspect, load, #parameters, #save, #train

Constructor Details

This class inherits a constructor from RubyZero::NN::Model

Instance Method Details

#forward(input, target) ⇒ Object 



3
4
5
6
7
8
9
10
 
# File 'lib/rubyzero/nn/losses/mse.rb', line 3

def forward(input, target)
    err = input - target
    err = err * err
    while err.shape.size > 1
        err = err.mean(axis:0)
    end
    return err
end