arnaucube/poulpy
1
0
Fork 0
mirror of https://github.com/arnaucube/poulpy.git synced 2026-02-10 21:26:41 +01:00
Code Issues Packages Projects Releases Wiki Activity

added tests for sampling (and indirectly stats)

Browse Source
This commit is contained in:
Jean-Philippe Bossuat
2025-04-28 09:02:42 +02:00
parent 78b6e9544d
commit 39bbe5b917
5 changed files with 107 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
base2k/src/stats.rs
View File

@@ -3,11 +3,16 @@ use rug::Float;
use rug::float::Round;
use rug::ops::{AddAssignRound, DivAssignRound, SubAssignRound};
impl VecZnx {
pub fn std(&self, poly_idx: usize, log_base2k: usize) -> f64 {
let prec: u32 = (self.cols() * log_base2k) as u32;
pub trait Stats {
/// Returns the standard devaition of the i-th polynomial.
fn std(&self, col_i: usize, log_base2k: usize) -> f64;
}
impl Stats for VecZnx {
fn std(&self, col_i: usize, log_base2k: usize) -> f64 {
let prec: u32 = (self.limbs() * log_base2k) as u32;
let mut data: Vec<Float> = (0..self.n()).map(|_| Float::with_val(prec, 0)).collect();
self.decode_vec_float(poly_idx, log_base2k, &mut data);
self.decode_vec_float(col_i, log_base2k, &mut data);
// std = sqrt(sum((xi - avg)^2) / n)
let mut avg: Float = Float::with_val(prec, 0);
data.iter().for_each(|x| {