prodef 0.2.2

A simple Rust crate for handling probability distributions, primarily intended for use with Bayesian inference.
Documentation 
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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71

use crate::{
    multivariate::MultivariateDensity,
    pytypes::{Float, PyUnivariate},
};
use nalgebra::Dyn;
use pyo3::{Bound, PyResult, exceptions::PyTypeError, prelude::*, types::PyList};

/// A multivariate density for use in Python.
#[derive(Clone)]
#[pyclass(from_py_object, name = "Multivariate")]
pub struct PyMultivariate {
    #[pyo3(get)]
    uvpdfs: Vec<PyUnivariate>,
}

impl PyMultivariate {
    /// Return a reference to the underlying univariate densities.
    pub fn uvpdfs(&self) -> &Vec<PyUnivariate> {
        &self.uvpdfs
    }
}

impl From<MultivariateDensity<Float, Dyn>> for PyMultivariate {
    /// Convert a [`MultivariateDensity`] with runtime dimensions to a [`PyMultivariate`].
    fn from(dist: MultivariateDensity<Float, Dyn>) -> Self {
        let uvpdfs = dist
            .marginals()
            .iter()
            .map(|uv| PyUnivariate::from(uv.clone()))
            .collect();
        Self { uvpdfs }
    }
}

#[pymethods]
impl PyMultivariate {
    #[new]
    /// Create a new [`PyMultivariate`] from a list of [`PyUnivariate`]s.
    pub fn new<'py>(priors: Bound<'py, PyList>) -> PyResult<Self> {
        let priors = priors
            .iter()
            .map(|p| match p.extract::<PyUnivariate>() {
                Ok(uvpdf) => Some(uvpdf.clone()),
                Err(_) => None,
            })
            .collect::<Vec<Option<PyUnivariate>>>();

        if priors.iter().any(|r| r.is_none()) {
            Err(PyTypeError::new_err(
                "failed to convert one of the members of the list argument to a PyUnivariate type",
            ))
        } else {
            Ok(Self {
                uvpdfs: priors.into_iter().map(|uvpdf| uvpdf.unwrap()).collect(),
            })
        }
    }

    /// Return the names of the underlying univariate densities.
    pub fn names(&self) -> Vec<String> {
        self.uvpdfs.iter().map(|u| u.name().to_string()).collect()
    }

    /// Return the type names of the underlying univariate densities.
    pub fn typenames(&self) -> Vec<String> {
        self.uvpdfs
            .iter()
            .map(|u| u.typename().to_string())
            .collect()
    }
}