1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42

//! # Probability Distributions Plugins
//!
//! This module containes plugins that simulate probability distributions on the nodes.
//! It also has trackers used to see when queries and sequences are out of distribution.

use super::*;
use rand::Rng;
use std::fmt::Debug;

mod diag_gaussian;
pub use diag_gaussian::*;

/*
There's an issue with rust-numpy and ndarray causing the linear algebra package for ndarray to fail. 

Temporary removal

mod svd_gaussian;
pub use svd_gaussian::*;
*/

use pointcloud::PointRef;

///
pub trait ContinousDistribution: Clone + 'static {
    /// Pass none if you want to test for a singleton, returns 0 if
    fn ln_pdf<T: PointRef>(&self, point: &T) -> Option<f64>;
    /// Samples a point from this distribution
    fn sample<R: Rng>(&self, rng: &mut R) -> Vec<f32>;

    /// Computes the KL divergence of two bucket probs.
    /// KL(self || other)
    /// Returns None if the support of the self is not a subset of the support of the other, or the calculation is undefined.
    fn kl_divergence(&self, other: &Self) -> Option<f64>;
}

///
pub trait ContinousBayesianDistribution: ContinousDistribution + Clone + 'static {
    /// Adds an observation to the distribution.
    /// This currently shifts the underlying parameters of the distribution rather than be tracked.
    fn add_observation<T: PointRef>(&mut self, point: &T);
}