d803bfe2b1
git-subtree-dir: vendor/ruvector git-subtree-split: b64c21726f2bb37286d9ee36a7869fef60cc6900
42 lines
984 B
Rust
42 lines
984 B
Rust
use hyperbolic_attention::prelude::*;
|
|
|
|
#[test]
|
|
fn debug_curvature_scaling() {
|
|
let x = vec![0.5, 0.0];
|
|
let y = vec![1.0, 0.0];
|
|
|
|
let d1 = poincare_distance(&x, &y, 1.0);
|
|
let d2 = poincare_distance(&x, &y, 2.0);
|
|
|
|
println!("Distance with K=1.0: {}", d1);
|
|
println!("Distance with K=2.0: {}", d2);
|
|
println!("d2 > d1: {}", d2 > d1);
|
|
}
|
|
|
|
#[test]
|
|
fn debug_exp_log() {
|
|
let x = vec![0.1, 0.2];
|
|
let y = vec![0.3, 0.1];
|
|
let k = 1.0;
|
|
|
|
println!("x: {:?}", x);
|
|
println!("y: {:?}", y);
|
|
|
|
let v = logarithmic_map(&x, &y, k);
|
|
println!("log_x(y) = v: {:?}", v);
|
|
|
|
let y_reconstructed = exponential_map(&x, &v, k);
|
|
println!("exp_x(v) = y': {:?}", y_reconstructed);
|
|
println!("Original y: {:?}", y);
|
|
|
|
for (i, (orig, recon)) in y.iter().zip(&y_reconstructed).enumerate() {
|
|
println!(
|
|
" y[{}]: {} vs {} (diff: {})",
|
|
i,
|
|
orig,
|
|
recon,
|
|
(orig - recon).abs()
|
|
);
|
|
}
|
|
}
|