skshapes.loss.nearest_neighbors.NearestNeighborsLoss
- class skshapes.loss.nearest_neighbors.NearestNeighborsLoss
Bases:
BaseLoss
Loss based on nearest neighbors for PolyData.
This class defines a loss corresponding to the nearest neighbors distance between the points of two PolyData objects. More precisely, for each point in the source PolyData, we compute the distance to its nearest neighbor in the target PolyData. The loss is then the average of these distances.
The distances are computed using the lazy tensor library pykeops : https://www.kernel-operations.io/keops/index.html
- __init__()
Methods
__init__
()