This class implements a layer that calculates a hinge
loss function for binary classification.
The function is calculated according to the formula:
loss = max(0, 1 - x * y)
where:
x
is the network response.y
is the correct class label (can be1
or-1
).
void SetLossWeight( float lossWeight );
Sets the multiplier for this function gradient during training. The default value is 1
. You may wish to change the default if you are using several loss functions in your network.
void SetMaxGradientValue( float maxValue );
Sets the upper limit for the absolute value of the function gradient. Whenever the gradient exceeds this limit its absolute value will be reduced to GetMaxGradientValue()
.
This layer has no trainable parameters.
The layer may have 2 to 3 inputs:
-
The network output for which you are calculating the loss function. It should contain the probability distribution for
BatchLength * BatchWidth * ListSize
objects overHeight * Width * Depth * Channels
classes. Each element should be greater or equal to0
. For each object, the sum of all elements overHeight * Width * Depth * Channels
dimension should be equal to1
. -
The class labels represented by a blob in one of the two formats:
- the blob contains
float
data, the dimensions are equal to the first input dimensions. It should be filled with zeros, and only the coordinate of the class to which the corresponding object from the first input belongs should be1
. - the blob contains
int
data withBatchLength
,BatchWidth
, andListSize
equal to these dimensions of the first input, and the other dimensions equal to1
. Each object in the blob contains the number of the class to which the corresponding object from the first input belongs.
- the blob contains
-
[Optional] The objects' weights. This input should have the same
BatchLength
,BatchWidth
, andListSize
dimensions as the first input.Height
,Width
,Depth
, andChannels
should be equal to1
.
This layer has no output.
float GetLastLoss() const;
Use this method to get the value of the loss function calculated on the network's last run.