swamauri_metric_wasserstein 0.9.1.dev3

Wasserstein metric plugin for Swarmauri vector comparisons.

Swamauri Metric Wasserstein

swamauri_metric_wasserstein provides a Swarmauri-compatible metric class for computing the first Wasserstein distance (Earth Mover's Distance) between numeric vectors.

Features

• Implements WassersteinMetric as a MetricBase plugin.
• Computes distance using a built-in pure-Python 1D Wasserstein implementation.
• Includes helper methods for batched distances and similarities.
• Supports Python 3.10, 3.11, and 3.12.

Installation

uv pip install swamauri_metric_wasserstein

pip install swamauri_metric_wasserstein

Usage

from swamauri_metric_wasserstein import WassersteinMetric
from swarmauri_standard.vectors.Vector import Vector

vector_a = Vector(value=[0.0, 1.0, 2.0])
vector_b = Vector(value=[0.0, 1.0, 2.0])

distance_calculator = WassersteinMetric()

distance = distance_calculator.distance(vector_a, vector_b)
similarity = distance_calculator.similarity(vector_a, vector_b)

print(f"Distance: {distance}")
print(f"Similarity: {similarity}")

Want to help?

If you want to contribute to swarmauri-sdk, read up on our guidelines for contributing that will help you get started.