gnu: Add python-imbalanced-learn.

* gnu/packages/machine-learning.scm (python-imbalanced-learn): New variable.
This commit is contained in:
Ricardo Wurmus 2022-12-07 10:48:06 +01:00
parent 91d4a78f41
commit df9a7c2032
No known key found for this signature in database
GPG key ID: 197A5888235FACAC

View file

@ -1181,6 +1181,52 @@ (define-public python-threadpoolctl
for scientific computing and data science (e.g. BLAS and OpenMP).")
(license license:bsd-3)))
(define-public python-imbalanced-learn
(package
(name "python-imbalanced-learn")
(version "0.9.1")
(source (origin
(method url-fetch)
(uri (pypi-uri "imbalanced-learn" version))
(sha256
(base32
"0qnrmysnqpc8ii1w5n8mci20gcjhmjr7khvk7f2apdbqc2pgf52f"))))
(build-system pyproject-build-system)
(arguments
(list
#:phases
'(modify-phases %standard-phases
(add-after 'unpack 'unbreak-tests
(lambda _
;; The doctests require tensorflow
(substitute* "setup.cfg"
(("--doctest-modules") ""))
;; Some tests require a home directory
(setenv "HOME" (getcwd))
;; We don't have keras
(delete-file "imblearn/keras/tests/test_generator.py")
;; We don't have tensorflow
(delete-file "imblearn/tensorflow/tests/test_generator.py"))))))
(propagated-inputs
(list python-joblib
python-numpy
python-scikit-learn
python-scipy
python-threadpoolctl))
(native-inputs
(list python-black
python-flake8
python-mypy
python-pandas
python-pytest
python-pytest-cov))
(home-page "https://github.com/scikit-learn-contrib/imbalanced-learn")
(synopsis "Toolbox for imbalanced dataset in machine learning")
(description "This is a Python package offering a number of re-sampling
techniques commonly used in datasets showing strong between-class imbalance.
It is compatible with @code{scikit-learn}.")
(license license:expat)))
(define-public python-pynndescent
(package
(name "python-pynndescent")