WassersteinWormhole
Embedding point-clouds by preserving Wasserstein distances with the Wormhole.
This implementation is written in Python3 and relies on FLAX, JAX, & JAX-OTT.
To install JAX, simply run the command:
pip install --upgrade pip install -U "jax[cuda12]”
And to install WassersteinWormhole along with the rest of the requirements:
pip install wassersteinwormhole
And running the Wormhole on your own set of point-clouds is as simple as:
from wassersteinwormhole import Wormhole
WormholeModel = Wormhole(point_clouds = point_clouds)
WormholeModel.train()
Embeddings = WormholeModel.encode(WormholeModel.point_clouds, WormholeModel.masks)
For more details, follow tutorial at https://wasserstienwormhole.readthedocs.io.
1 maintainer
Metadata
1 maintainer
Release files for wassersteinwormhole 0.3.7
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| wassersteinwormhole-0.3.7.tar.gz | 18.8 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| wassersteinwormhole-0.3.7-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 42.6 kB
Release files / wassersteinwormhole-0.3.7.tar.gz
| Download URL | wassersteinwormhole-0.3.7.tar.gz |
|---|---|
| Size | 18.8 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
aa94b49b591e152274388648331a3df09ead55f6ac4a21612d37727cc6ab1d60
|
|
BLAKE2b-256 checksum How to use checksums |
e6d9e9c74c3102be642a21c0e17722e50367278754d9605c1308af02f3d7cd27
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
poetry/1.5.1 CPython/3.14.3 Linux/6.14.0-1017-azure
|
Release files / wassersteinwormhole-0.3.7-py3-none-any.whl
| Download URL | wassersteinwormhole-0.3.7-py3-none-any.whl |
|---|---|
| Size | 23.8 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
9458fdb9d6c8563592295d54f9e32b421282ec59c99ef77e406bff546d250afd
|
|
BLAKE2b-256 checksum How to use checksums |
cebd3792dd04bc55d6e6c92ce16d96bc5db5cb688e3767fa8342391c49be3aec
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
poetry/1.5.1 CPython/3.14.3 Linux/6.14.0-1017-azure
|
