gnu: Add python-cleanlab.

* gnu/packages/machine-learning.scm (python-cleanlab): New variable.
This commit is contained in:
Ricardo Wurmus 2022-12-06 15:14:46 +01:00
parent 2c8a53cb6b
commit eb23379828
No known key found for this signature in database
GPG key ID: 197A5888235FACAC

View file

@ -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")