yamle.losses package

yamle.losses.loss_factory(loss_type=None)

This function is used to create a loss instance based on the loss type.

Parameters:

loss_type (str) – The type of loss to create.

Return type:

Type[Callable]

Submodules

• yamle.losses.classification module
  • one_hot()
  • CrossEntropyLoss
    • CrossEntropyLoss.tasks
    • CrossEntropyLoss.add_specific_args()
  • TextCrossEntropyLoss
    • TextCrossEntropyLoss.tasks
• yamle.losses.contrastive module
  • NoiseContrastiveEstimatorLoss
    • NoiseContrastiveEstimatorLoss.add_specific_args()
• yamle.losses.evidential_regression module
  • NIG_NLL
    • NIG_NLL.forward()
    • NIG_NLL.training
  • NIG_Reg
    • NIG_Reg.forward()
    • NIG_Reg.training
  • EvidentialRegressionLoss
    • EvidentialRegressionLoss.tasks
• yamle.losses.loss module
  • BaseLoss
    • BaseLoss.tasks
    • BaseLoss.set_reduction_per_member()
    • BaseLoss.set_reduction_per_sample()
    • BaseLoss.set_reduction_per_feature()
    • BaseLoss.add_specific_args()
  • DummyLoss
• yamle.losses.regression module
  • GaussianNegativeLogLikelihoodLoss
    • GaussianNegativeLogLikelihoodLoss.tasks
    • GaussianNegativeLogLikelihoodLoss.add_specific_args()
  • MeanSquaredError
    • MeanSquaredError.tasks
    • MeanSquaredError.add_specific_args()
  • QuantileRegressionLoss
    • QuantileRegressionLoss.tasks
    • QuantileRegressionLoss.add_specific_args()
• yamle.losses.segmentation module
  • SoftIntersectionOverUnionLoss
    • SoftIntersectionOverUnionLoss.add_specific_args()
  • FocalLoss
    • FocalLoss.add_specific_args()