• Dennis Francis's avatar
    feature_addition : Mean squared loss layer for multiple output (#404) (#427) · 4dbe3337
    Dennis Francis authored
    * feature_addition : Mean squared loss layer for multiple output (#404)
    
    * Added loss_mean_squared_multioutput layer to support multiple outputs.
    * Also added a corresponding test case to test a single variable regression
      with multiple outputs.
    
    * Added error checks on truth argument
    
    Added assert statements to check that truth argument in
    compute_loss_value_and_gradient() method contains matrices
    of correct dimension relative to the output tensor's size.
    Also the requirements on argument truth to the abstract
    documentation.
    4dbe3337
loss_abstract.h 31.1 KB