Skip to main content

unshift_logits

Function unshift_logits 

Source
pub fn unshift_logits(
    logits: &[f64],
    shift: usize,
    n_classes: usize,
) -> Vec<f64>
Expand description

Undoes a member’s class-label shift: that member was fed (y_true + shift) % n_classes as training labels, so its output logit at position (c + shift) % n_classes is the prediction for original class c.