ryan77627/guix
1
0
Fork 0
mirror of https://git.in.rschanz.org/ryan77627/guix.git synced 2024-12-24 21:38:07 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

gnu: Add python-captum.

Browse source 
* gnu/packages/machine-learning.scm (python-captum): New variable.
This commit is contained in:
Ricardo Wurmus 2023-07-10 13:13:11 +02:00
parent e3d9d896b5
commit d0296970fb
No known key found for this signature in database
GPG key ID: 197A5888235FACAC
1 changed files with 45 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

45
gnu/packages/machine-learning.scm
View file

@ -3868,6 +3868,51 @@ (define-public python-lightning-utilities
Actions for the Lightning suite of libraries.")
(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
(package
(name "python-readchar")