[Date Prev][Date Next][Thread Prev][Thread Next][Date Index][Thread Index]
09/09: gnu: Add python-torchmetrics.
From: |
guix-commits |
Subject: |
09/09: gnu: Add python-torchmetrics. |
Date: |
Wed, 10 May 2023 02:40:04 -0400 (EDT) |
rekado pushed a commit to branch wip-pytorch-lightning
in repository guix.
commit 46c7243e69128dbac50bd4f89c22e6a7f8557387
Author: Ricardo Wurmus <rekado@elephly.net>
AuthorDate: Wed May 10 08:37:04 2023 +0200
gnu: Add python-torchmetrics.
* gnu/packages/machine-learning.scm (python-torchmetrics): New variable.
---
gnu/packages/machine-learning.scm | 48 +++++++++++++++++++++++++++++++++++++++
1 file changed, 48 insertions(+)
diff --git a/gnu/packages/machine-learning.scm
b/gnu/packages/machine-learning.scm
index de49e25c28..a3880247a7 100644
--- a/gnu/packages/machine-learning.scm
+++ b/gnu/packages/machine-learning.scm
@@ -3465,6 +3465,54 @@ AI services.")
Actions for the Lightning suite of libraries.")
(license license:asl2.0)))
+(define-public python-torchmetrics
+ (package
+ (name "python-torchmetrics")
+ (version "0.11.4")
+ (source (origin
+ (method url-fetch)
+ (uri (pypi-uri "torchmetrics" version))
+ (sha256
+ (base32
+ "150lcy6c20n42rwxl4d3m1b8s4js9ddds5wh3685vmjdnha5mr0z"))))
+ (build-system pyproject-build-system)
+ (propagated-inputs
+ (list python-numpy python-packaging python-pytorch
+ python-typing-extensions))
+ (native-inputs
+ (list python-cloudpickle
+ python-coverage
+ python-fire
+ python-mir-eval
+ python-mypy
+ python-pandas
+ python-psutil
+ python-pytest
+ python-pytest-cov
+ python-pytest-doctestplus
+ python-pytest-rerunfailures
+ python-pytest-timeout
+ python-requests
+ python-scikit-image
+ python-scikit-learn
+ python-scipy
+ python-types-protobuf
+ python-types-setuptools))
+ (home-page "https://github.com/Lightning-AI/metrics")
+ (synopsis "Machine learning metrics for PyTorch applications")
+ (description "TorchMetrics is a collection of 100+ PyTorch metrics
+implementations and an easy-to-use API to create custom metrics. It offers:
+
+@itemize
+@item A standardized interface to increase reproducibility
+@item Reduces boilerplate
+@item Automatic accumulation over batches
+@item Metrics optimized for distributed-training
+@item Automatic synchronization between multiple devices
+@end itemize
+")
+ (license license:asl2.0)))
+
;; Keep this in sync with python-pytorch
(define-public python-torchvision
(package
- branch wip-pytorch-lightning created (now 46c7243e69), guix-commits, 2023/05/10
- 01/09: gnu: python-pytorch: Update to 1.13.1., guix-commits, 2023/05/10
- 03/09: gnu: python-traitlets: Update to 5.9.0., guix-commits, 2023/05/10
- 08/09: gnu: Add python-types-setuptools., guix-commits, 2023/05/10
- 05/09: gnu: Add python-lightning-cloud., guix-commits, 2023/05/10
- 06/09: gnu: Add python-lightning-utilities., guix-commits, 2023/05/10
- 04/09: gnu: python-pytorch-for-r-torch: Define as alias for python-pytorch., guix-commits, 2023/05/10
- 07/09: gnu: Add python-mir-eval., guix-commits, 2023/05/10
- 02/09: gnu: Add python-deepdiff., guix-commits, 2023/05/10
- 09/09: gnu: Add python-torchmetrics.,
guix-commits <=