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.