Loss function to regularize the neural model¶
torch_explain.nn.functional.loss
- torch_explain.nn.functional.loss.entropy_logic_loss(model: torch.nn.modules.module.Module)¶
Entropy loss function to get simple logic explanations.
- Parameters
model – pytorch model.
- Returns
entropy loss.
- torch_explain.nn.functional.loss.l1_loss(model: torch.nn.modules.module.Module)¶
L1 loss function to get simple logic explanations.
- Parameters
model – pytorch model.
- Returns
L1 loss.