diff options
| author | Ricardo Wurmus <rekado@elephly.net> | 2023-05-10 08:37:04 +0200 |
|---|---|---|
| committer | Ricardo Wurmus <rekado@elephly.net> | 2023-05-10 19:27:07 +0200 |
| commit | 8f97d96ddf8c280a6ce6a994f71b670d60f21cb7 (patch) | |
| tree | 237a008de521bfff356e4dfebbd2f1fd92c47988 /gnu/packages | |
| parent | 1aa1a6a8161e59d909381b9bad838df74e072138 (diff) | |
gnu: Add python-torchmetrics.
* gnu/packages/machine-learning.scm (python-torchmetrics): New variable.
Diffstat (limited to 'gnu/packages')
| -rw-r--r-- | gnu/packages/machine-learning.scm | 48 |
1 files changed, 48 insertions, 0 deletions
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 |