Loss functions to regularize the neural model

torch_explain.nn.functional.loss

torch_explain.nn.functional.loss.entropy_logic_loss(model: Module)

Entropy loss function to get simple logic explanations.

Parameters:

model – pytorch model.

Returns:

entropy loss.

torch_explain.nn.functional.loss.l1_loss(model: Module)

L1 loss function to get simple logic explanations.

Parameters:

model – pytorch model.

Returns:

L1 loss.