Struct WassersteinKernelSampler 

pub struct WassersteinKernelSampler { /* private fields */ }

Wasserstein kernel approximation using random features

This implements approximations to the Wasserstein kernel which computes optimal transport distances between empirical distributions. WassersteinKernelSampler

Implementations

impl WassersteinKernelSampler

pub fn new(n_components: usize) -> Self

Create a new Wasserstein kernel approximation sampler

Arguments

• n_components - Number of random features to generate

Examples

use sklears_kernel_approximation::WassersteinKernelSampler;
let sampler = WassersteinKernelSampler::new(100);

pub fn ground_metric(self, metric: GroundMetric) -> Self

Set the ground metric for optimal transport

pub fn epsilon(self, epsilon: f64) -> Self

Set the regularization parameter for Sinkhorn divergence

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

Set the maximum number of iterations for Sinkhorn

pub fn tolerance(self, tolerance: f64) -> Self

Set the convergence tolerance

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

Set the random state for reproducibility

pub fn transport_method(self, method: TransportMethod) -> Self

Set the transport approximation method

Trait Implementations

impl Clone for WassersteinKernelSampler

fn clone(&self) -> WassersteinKernelSampler

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for WassersteinKernelSampler

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

Formats the value using the given formatter.

impl Default for WassersteinKernelSampler

fn default() -> Self

Returns the “default value” for a type.

impl Fit<ArrayBase<OwnedRepr<f64>, Dim<[usize; 2]>>, ()> for WassersteinKernelSampler

type Fitted = FittedWassersteinSampler

The fitted model type

fn fit(self, x: &Array2<f64>, _y: &()) -> SklResult<Self::Fitted>

Fit the model to the provided data with validation

fn fit_with_validation( self, x: &X, y: &Y, _x_val: Option<&X>, _y_val: Option<&Y>, ) -> Result<(Self::Fitted, FitMetrics), SklearsError>

where Self: Sized,

Fit with custom validation and early stopping