skshapes.loss.optimal_transport.OptimalTransportLoss

class skshapes.loss.optimal_transport.OptimalTransportLoss(loss='sinkhorn', **kwargs)

Bases: BaseLoss

Loss based on optimal transport for PolyData.

This class defines a loss based on optimal transport for PolyData. More precisely, it initializes a SamplesLoss object from the geomloss library. See its [documentation]((https://www.kernel-operations.io/geomloss/)) for more details. The default loss is the Sinkhorn loss.

Parameters:

loss (Literal['sinkhorn', 'hausdorff', 'energy', 'gaussian', 'laplacian']) – The loss function to compute. Supported values are “sinkhorn”, “hausdorff”, “energy”, “gaussian” and “laplacian”.
**kwargs – additional arguments passed to the geomloss.SamplesLoss object.

__init__(loss='sinkhorn', **kwargs)

Methods

__init__([loss])