mirror of
https://git.in.rschanz.org/ryan77627/guix.git
synced 2024-12-24 21:38:07 -05:00
gnu: Add python-captum.
* gnu/packages/machine-learning.scm (python-captum): New variable.
This commit is contained in:
parent
e3d9d896b5
commit
d0296970fb
1 changed files with 45 additions and 0 deletions
|
@ -3868,6 +3868,51 @@ (define-public python-lightning-utilities
|
||||||
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-captum
|
||||||
|
(package
|
||||||
|
(name "python-captum")
|
||||||
|
(version "0.6.0")
|
||||||
|
(source (origin
|
||||||
|
(method git-fetch)
|
||||||
|
(uri (git-reference
|
||||||
|
(url "https://github.com/pytorch/captum")
|
||||||
|
(commit (string-append "v" version))))
|
||||||
|
(file-name (git-file-name name version))
|
||||||
|
(sha256
|
||||||
|
(base32
|
||||||
|
"1h4n91ivhjxm6wj0vgqpfss2dmq4sjcp0appd08cd5naisabjyb5"))))
|
||||||
|
(build-system pyproject-build-system)
|
||||||
|
(arguments
|
||||||
|
(list
|
||||||
|
#:test-flags
|
||||||
|
'(list "-k"
|
||||||
|
;; These two tests (out of more than 1000 tests) fail because of
|
||||||
|
;; accuracy problems.
|
||||||
|
"not test_softmax_classification_batch_multi_target\
|
||||||
|
and not test_softmax_classification_batch_zero_baseline")))
|
||||||
|
(propagated-inputs (list python-matplotlib python-numpy python-pytorch))
|
||||||
|
(native-inputs (list jupyter
|
||||||
|
python-annoy
|
||||||
|
python-black
|
||||||
|
python-flake8
|
||||||
|
python-flask
|
||||||
|
python-flask-compress
|
||||||
|
python-ipython
|
||||||
|
python-ipywidgets
|
||||||
|
python-mypy
|
||||||
|
python-parameterized
|
||||||
|
python-pytest
|
||||||
|
python-pytest-cov
|
||||||
|
python-scikit-learn))
|
||||||
|
(home-page "https://captum.ai")
|
||||||
|
(synopsis "Model interpretability for PyTorch")
|
||||||
|
(description "Captum is a model interpretability and understanding library
|
||||||
|
for PyTorch. Captum contains general purpose implementations of integrated
|
||||||
|
gradients, saliency maps, smoothgrad, vargrad and others for PyTorch models.
|
||||||
|
It has quick integration for models built with domain-specific libraries such
|
||||||
|
as torchvision, torchtext, and others.")
|
||||||
|
(license license:bsd-3)))
|
||||||
|
|
||||||
(define-public python-readchar
|
(define-public python-readchar
|
||||||
(package
|
(package
|
||||||
(name "python-readchar")
|
(name "python-readchar")
|
||||||
|
|
Loading…
Reference in a new issue