summaryrefslogtreecommitdiff
path: root/gnu/packages/machine-learning.scm
diff options
context:
space:
mode:
authorRicardo Wurmus <rekado@elephly.net>2025-01-22 15:44:12 +0100
committerRicardo Wurmus <rekado@elephly.net>2025-01-22 15:44:12 +0100
commit9fa107c774017e35d2b0f05c2032ba520b9c0218 (patch)
tree75c3d5bac493704276380f3998d9b00404bcd9e2 /gnu/packages/machine-learning.scm
parent803befde95b6315199423b9aaca8054d5bf91d73 (diff)
gnu: Add python-pot.
* gnu/packages/machine-learning.scm (python-pot): New variable. Change-Id: I2a844a41031ce80d46a3ffe03ead9652d5a87348
Diffstat (limited to 'gnu/packages/machine-learning.scm')
-rw-r--r--gnu/packages/machine-learning.scm30
1 files changed, 30 insertions, 0 deletions
diff --git a/gnu/packages/machine-learning.scm b/gnu/packages/machine-learning.scm
index ecc9aa3a19..4d1d49553a 100644
--- a/gnu/packages/machine-learning.scm
+++ b/gnu/packages/machine-learning.scm
@@ -1092,6 +1092,36 @@ It currently houses implementations of
")
(license license:expat))) ; MIT License
+(define-public python-pot
+ (package
+ (name "python-pot")
+ (version "0.9.5")
+ (source
+ (origin
+ (method url-fetch)
+ (uri (pypi-uri "pot" version))
+ (sha256
+ (base32 "0hk0dmjgnpwka0a7gyzrcq155wzlvzcrsav3qaizyg0wymzywi4n"))
+ (snippet '(delete-file "ot/lp/emd_wrap.cpp"))))
+ (build-system pyproject-build-system)
+ (propagated-inputs
+ (list python-autograd
+ python-numpy
+ python-pytorch
+ python-pytorch-geometric
+ python-pymanopt
+ python-scikit-learn
+ python-scipy))
+ (native-inputs (list python-cython
+ python-setuptools
+ python-wheel))
+ (home-page "https://github.com/PythonOT/POT")
+ (synopsis "Python Optimal Transport Library")
+ (description "This Python library provides several solvers for
+optimization problems related to Optimal Transport for signal, image
+processing and machine learning.")
+ (license license:expat)))
+
(define-public python-pymanopt
(package
(name "python-pymanopt")