Fix snr helper bug

Ledger-Donjon · Oct 31, 2024 · 7c1d996 · 7c1d996
1 parent 160c550
commit 7c1d996
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/leakage_detection.rs b/src/leakage_detection.rs
@@ -63,7 +63,7 @@ where
             || SnrProcessor::new(traces.shape()[1], classes),
             |mut snr, (batch_idx, trace_batch)| {
                 for i in 0..trace_batch.shape()[0] {
-                    snr.process(trace_batch.row(i), get_class(batch_idx + i));
+                    snr.process(trace_batch.row(i), get_class(batch_idx * batch_size + i));
                 }
                 snr
             },