kernel_density_estimation/kernel/
tricube.rs

1//! Tricube kernel function.
2
3use crate::float::{float, KDEFloat};
4use crate::kernel::Kernel;
5
6/// Tricube kernel function.
7#[derive(Clone, Copy, Debug)]
8pub struct Tricube;
9
10impl<F: KDEFloat> Kernel<F> for Tricube {
11    fn pdf(&self, x: F) -> F {
12        let abs_x = x.abs();
13        if abs_x <= F::one() {
14            float!(70.0 / 81.0) * (F::one() - abs_x.powi(3)).powi(3)
15        } else {
16            F::zero()
17        }
18    }
19}
20
21#[cfg(test)]
22mod tests {
23    use super::Tricube;
24    use crate::kernel::Kernel;
25    use approx::*;
26
27    #[test]
28    fn tricube() {
29        let kernel = Tricube;
30
31        let x = 0.0;
32        let res = kernel.pdf(x);
33        assert_relative_eq!(res, 0.86419, epsilon = 1.0e-5);
34
35        let x = -1.0;
36        let res = kernel.pdf(x);
37        assert_relative_eq!(res, 0.0, epsilon = 1.0e-5);
38
39        let x = 1.0;
40        let res = kernel.pdf(x);
41        assert_relative_eq!(res, 0.0, epsilon = 1.0e-5);
42
43        let x = 0.5;
44        let res = kernel.pdf(x);
45        assert_relative_eq!(res, 0.57895, epsilon = 1.0e-5);
46    }
47}