Distributions

quantflow.dists.Distribution pydantic-model

Bases: BaseModel, ABC

Base class for distributions.

sample abstractmethod

sample(size=1)

Draw random samples from the distribution.

PARAMETER	DESCRIPTION
size	Number of samples to draw. TYPE: int DEFAULT: 1

Source code in quantflow/dists/base.py

@abstractmethod
def sample(
    self,
    size: Annotated[int, Doc("Number of samples to draw.")] = 1,
) -> FloatArray:
    """Draw random samples from the distribution."""
    ...

quantflow.dists.MvDistribution pydantic-model

Bases: Distribution

sample abstractmethod

sample(size=1)

Draw random samples from the distribution.

PARAMETER	DESCRIPTION
size	Number of samples to draw. TYPE: int DEFAULT: 1

Source code in quantflow/dists/base.py

@abstractmethod
def sample(
    self,
    size: Annotated[int, Doc("Number of samples to draw.")] = 1,
) -> FloatArray:
    """Draw random samples from the distribution."""
    ...

mean_and_cov abstractmethod

mean_and_cov()

Mean vector and covariance matrix of the distribution.

Source code in quantflow/dists/base.py

@abstractmethod
def mean_and_cov(self) -> MeanAndCov:
    """Mean vector and covariance matrix of the distribution."""
    ...

quantflow.dists.MeanAndCov

Bases: NamedTuple

Mean vector and covariance matrix of a distribution.

mean instance-attribute

mean

cov instance-attribute

cov

quantflow.dists.MvNormal pydantic-model

Bases: MvDistribution

Multivariate normal distribution \(\mathcal{N}(\mu, \Sigma)\).

Fields:

• mean (FloatArray)
• cov (FloatArray)

mean pydantic-field

mean

cov pydantic-field

cov

sample

sample(size=1)

Source code in quantflow/dists/mv_normal.py

def sample(self, size: int = 1) -> FloatArray:
    return np.asarray(
        multivariate_normal(mean=self.mean, cov=self.cov).rvs(size=size)
    )

log_pdf

log_pdf(x)

Source code in quantflow/dists/mv_normal.py

def log_pdf(self, x: FloatArray) -> FloatArray:
    return np.asarray(multivariate_normal(mean=self.mean, cov=self.cov).logpdf(x))

mean_and_cov

mean_and_cov()

Source code in quantflow/dists/mv_normal.py

def mean_and_cov(self) -> MeanAndCov:
    return MeanAndCov(self.mean, self.cov)