Struct WassersteinConfig 

pub struct WassersteinConfig {
    pub num_projections: usize,
    pub regularization: f64,
    pub max_iterations: usize,
    pub threshold: f64,
    pub p: f64,
    pub seed: Option<u64>,
}

Configuration for Wasserstein distance computation

Fields

num_projections: usize

Number of random projections for Sliced Wasserstein

regularization: f64

Regularization parameter for Sinkhorn (epsilon)

max_iterations: usize

Maximum iterations for Sinkhorn

threshold: f64

Convergence threshold for Sinkhorn

p: f64

Power p for Wasserstein-p distance

seed: Option<u64>

Random seed for reproducibility

Implementations

impl WassersteinConfig

pub fn new() -> Self

Create a new configuration with default values

pub fn with_projections(self, n: usize) -> Self

Set the number of random projections

pub fn with_regularization(self, eps: f64) -> Self

Set the regularization parameter

pub fn with_max_iterations(self, max_iter: usize) -> Self

Set the maximum iterations

pub fn with_threshold(self, threshold: f64) -> Self

Set the convergence threshold

pub fn with_power(self, p: f64) -> Self

Set the Wasserstein power

pub fn with_seed(self, seed: u64) -> Self

Set the random seed

pub fn validate(&self) -> Result<()>

Validate the configuration

Trait Implementations

impl Clone for WassersteinConfig

fn clone(&self) -> WassersteinConfig

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for WassersteinConfig

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Default for WassersteinConfig

fn default() -> Self

Returns the “default value” for a type.