mirror of
https://git.in.rschanz.org/ryan77627/guix.git
synced 2024-12-24 05:18:07 -05:00
gnu: Add python-cleanlab.
* gnu/packages/machine-learning.scm (python-cleanlab): New variable.
This commit is contained in:
parent
2c8a53cb6b
commit
eb23379828
1 changed files with 46 additions and 0 deletions
|
@ -43,6 +43,7 @@ (define-module (gnu packages machine-learning)
|
|||
#:use-module (guix build-system cmake)
|
||||
#:use-module (guix build-system gnu)
|
||||
#:use-module (guix build-system ocaml)
|
||||
#:use-module (guix build-system pyproject)
|
||||
#:use-module (guix build-system python)
|
||||
#:use-module (guix build-system r)
|
||||
#:use-module (guix build-system trivial)
|
||||
|
@ -1283,6 +1284,51 @@ (define-public python-scikit-rebate
|
|||
standard feature selection algorithms.")
|
||||
(license license:expat)))
|
||||
|
||||
(define-public python-cleanlab
|
||||
(package
|
||||
(name "python-cleanlab")
|
||||
(version "2.2.0")
|
||||
;; The version on pypi does not come with tests.
|
||||
(source (origin
|
||||
(method git-fetch)
|
||||
(uri (git-reference
|
||||
(url "https://github.com/cleanlab/cleanlab")
|
||||
(commit (string-append "v" version))))
|
||||
(file-name (git-file-name name version))
|
||||
(sha256
|
||||
(base32
|
||||
"00dqhxpwg781skknw943ynll2s44g4j125dx8aapk1d5d71sbzqy"))))
|
||||
(build-system pyproject-build-system)
|
||||
(arguments
|
||||
(list
|
||||
#:phases
|
||||
'(modify-phases %standard-phases
|
||||
(add-after 'unpack 'disable-bad-tests
|
||||
(lambda _
|
||||
;; XXX This requires pytest lazy_fixture
|
||||
(delete-file "tests/test_multilabel_classification.py")
|
||||
;; Requires tensorflow
|
||||
(delete-file "tests/test_frameworks.py")
|
||||
;; Tries to download datasets from the internet at runtime.
|
||||
(delete-file "tests/test_dataset.py"))))))
|
||||
(propagated-inputs
|
||||
(list python-numpy
|
||||
python-pandas
|
||||
python-scikit-learn
|
||||
python-termcolor
|
||||
python-tqdm))
|
||||
(native-inputs
|
||||
(list python-pytest
|
||||
python-pytorch
|
||||
python-torchvision))
|
||||
(home-page "https://cleanlab.ai")
|
||||
(synopsis "Automatically find and fix dataset issues")
|
||||
(description
|
||||
"cleanlab automatically finds and fixes errors in any ML dataset. This
|
||||
data-centric AI package facilitates machine learning with messy, real-world
|
||||
data by providing clean labels during training.")
|
||||
(license license:agpl3+)))
|
||||
|
||||
(define-public python-cmaes
|
||||
(package
|
||||
(name "python-cmaes")
|
||||
|
|
Loading…
Reference in a new issue