gnu: Add python-torchmetrics.
* gnu/packages/machine-learning.scm (python-torchmetrics): New variable.
This commit is contained in:
parent
1aa1a6a816
commit
8f97d96ddf
1 changed files with 48 additions and 0 deletions
|
@ -3465,6 +3465,54 @@ AI services.")
|
||||||
Actions for the Lightning suite of libraries.")
|
Actions for the Lightning suite of libraries.")
|
||||||
(license license:asl2.0)))
|
(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
|
;; Keep this in sync with python-pytorch
|
||||||
(define-public python-torchvision
|
(define-public python-torchvision
|
||||||
(package
|
(package
|
||||||
|
|
Reference in a new issue