Struct ConstraintLoss 

pub struct ConstraintLoss { /* private fields */ }

Constraint-aware loss wrapper

Bridges kizzasi-logic constraints with candle tensor operations. Allows combining task loss with constraint violations for constrained optimization.

Examples

use kizzasi_core::{ConstraintLoss, Loss};

let constraint_loss = ConstraintLoss::new(0.1);

// In training loop:
let task_loss = Loss::mse(&predictions, &targets)?;
let total_loss = constraint_loss.compute(&task_loss, &predictions, |pred| {
    // Compute constraint violation from prediction
    Ok(0.0)
})?;

Implementations

impl ConstraintLoss

pub fn new(constraint_weight: f32) -> Self

Create a new constraint-aware loss

pub fn compute<F>( &self, task_loss: &Tensor, prediction: &Tensor, constraint_fn: F, ) -> CoreResult<Tensor>

where F: Fn(&Tensor) -> CoreResult<f32>,

Compute combined loss: task_loss + constraint_weight * constraint_penalty

Arguments

• task_loss - Base task loss (MSE, MAE, etc.)
• prediction - Model prediction tensor
• constraint_fn - Function that computes constraint violation from prediction