Skip to content

Add BCE loss function

Olivier Antoni requested to merge oantoni/aidge_learning:BCE_loss into dev

Context

Add Binary Cross Entropy (BCE) loss function for classification.

Detailed major modifications

BCE loss function is computed using a graph in order to not be backend dependant.

A unit test is added.

Merge request reports

Copyright © Eclipse Foundation, Inc. All Rights Reserved.     Privacy Policy | Terms of Use | Copyright Agent