The dice loss performance function measures the network's performance according to the dice coefficient between the input and target distributions. More...
Public Member Functions | |
DiceLoss (const double smooth=1) | |
Create the DiceLoss object. More... | |
template < typename PredictionType , typename TargetType , typename LossType > | |
void | Backward (const PredictionType &prediction, const TargetType &target, LossType &loss) |
Ordinary feed backward pass of a neural network. More... | |
template < typename PredictionType , typename TargetType > | |
PredictionType::elem_type | Forward (const PredictionType &prediction, const TargetType &target) |
Computes the dice loss function. More... | |
OutputDataType & | OutputParameter () const |
Get the output parameter. More... | |
OutputDataType & | OutputParameter () |
Modify the output parameter. More... | |
template < typename Archive > | |
void | serialize (Archive &ar, const uint32_t) |
Serialize the layer. More... | |
double | Smooth () const |
Get the smooth. More... | |
double & | Smooth () |
Modify the smooth. More... | |
The dice loss performance function measures the network's performance according to the dice coefficient between the input and target distributions.
For more information see the following.
InputDataType | Type of the input data (arma::colvec, arma::mat, arma::sp_mat or arma::cube). |
OutputDataType | Type of the output data (arma::colvec, arma::mat, arma::sp_mat or arma::cube). |
Definition at line 50 of file dice_loss.hpp.
DiceLoss | ( | const double | smooth = 1 | ) |
Create the DiceLoss object.
smooth | The Laplace smoothing parameter. |
void Backward | ( | const PredictionType & | prediction, |
const TargetType & | target, | ||
LossType & | loss | ||
) |
Ordinary feed backward pass of a neural network.
prediction | Predictions used for evaluating the specified loss function. |
target | The target vector. |
loss | The calculated error. |
PredictionType::elem_type Forward | ( | const PredictionType & | prediction, |
const TargetType & | target | ||
) |
Computes the dice loss function.
prediction | Predictions used for evaluating the specified loss function. |
target | The target vector. |
|
inline |
Get the output parameter.
Definition at line 85 of file dice_loss.hpp.
|
inline |
Modify the output parameter.
Definition at line 87 of file dice_loss.hpp.
void serialize | ( | Archive & | ar, |
const uint32_t | |||
) |
Serialize the layer.
Referenced by DiceLoss< InputDataType, OutputDataType >::Smooth().
|
inline |
Get the smooth.
Definition at line 90 of file dice_loss.hpp.
|
inline |
Modify the smooth.
Definition at line 92 of file dice_loss.hpp.
References DiceLoss< InputDataType, OutputDataType >::serialize().