Commented out some prints in chi2 compare matrices

pull/68/head
Meliurwen 2 years ago
parent 832922922a
commit 0ae2168a94
Signed by: meliurwen
GPG Key ID: 818A8B35E9F1CE10
  1. 18
      reCTBN/src/structure_learning/hypothesis_test.rs

@ -97,7 +97,7 @@ impl ChiSquare {
let n = K.len(); let n = K.len();
K.into_shape((n, 1)).unwrap() K.into_shape((n, 1)).unwrap()
}; };
println!("K: {:?}", K); //println!("K: {:?}", K);
let L = 1.0 / &K; let L = 1.0 / &K;
// ===== 2 // ===== 2
// \ (K . M - L . M) // \ (K . M - L . M)
@ -108,18 +108,18 @@ impl ChiSquare {
// x'ϵVal /X \ // x'ϵVal /X \
// \ i/ // \ i/
let mut X_2 = (&K * &M2 - &L * &M1).mapv(|a| a.powi(2)) / (&M2 + &M1); let mut X_2 = (&K * &M2 - &L * &M1).mapv(|a| a.powi(2)) / (&M2 + &M1);
println!("M1: {:?}", M1); //println!("M1: {:?}", M1);
println!("M2: {:?}", M2); //println!("M2: {:?}", M2);
println!("L*M1: {:?}", (L * &M1)); //println!("L*M1: {:?}", (L * &M1));
println!("K*M2: {:?}", (K * &M2)); //println!("K*M2: {:?}", (K * &M2));
println!("X_2: {:?}", X_2); //println!("X_2: {:?}", X_2);
X_2.diag_mut().fill(0.0); X_2.diag_mut().fill(0.0);
let X_2 = X_2.sum_axis(Axis(1)); let X_2 = X_2.sum_axis(Axis(1));
let n = ChiSquared::new((X_2.dim() - 1) as f64).unwrap(); let n = ChiSquared::new((X_2.dim() - 1) as f64).unwrap();
println!("CHI^2: {:?}", n); //println!("CHI^2: {:?}", n);
println!("CHI^2 CDF: {:?}", X_2.mapv(|x| n.cdf(x))); //println!("CHI^2 CDF: {:?}", X_2.mapv(|x| n.cdf(x)));
let ret = X_2.into_iter().all(|x| n.cdf(x) < (1.0 - self.alpha)); let ret = X_2.into_iter().all(|x| n.cdf(x) < (1.0 - self.alpha));
println!("test: {:?}", ret); //println!("test: {:?}", ret);
ret ret
} }
} }

Loading…
Cancel
Save