summaryrefslogtreecommitdiff
path: root/gnu/packages/machine-learning.scm
diff options
context:
space:
mode:
Diffstat (limited to 'gnu/packages/machine-learning.scm')
-rw-r--r--gnu/packages/machine-learning.scm51
1 files changed, 51 insertions, 0 deletions
diff --git a/gnu/packages/machine-learning.scm b/gnu/packages/machine-learning.scm
index 93eee3de88..4dfeddedb0 100644
--- a/gnu/packages/machine-learning.scm
+++ b/gnu/packages/machine-learning.scm
@@ -1189,6 +1189,57 @@ Models, is a program for performing both single-SNP and SNP-set genome-wide
association studies (GWAS) on extremely large data sets.")
(license license:asl2.0)))
+(define-public python-hyperopt
+ (package
+ (name "python-hyperopt")
+ (version "0.2.5")
+ (source
+ (origin
+ (method url-fetch)
+ (uri (pypi-uri "hyperopt" version))
+ (sha256
+ (base32 "1k4ma8ci0bxghw7g4ms944zak1pi83yv2d6bxd7fcslm1zalfq5w"))))
+ (build-system python-build-system)
+ (arguments
+ `(#:phases
+ (modify-phases %standard-phases
+ (replace 'check
+ (lambda* (#:key inputs outputs tests? #:allow-other-keys)
+ (when tests?
+ (add-installed-pythonpath inputs outputs)
+ (invoke "python" "-m" "pytest" "--ignore"
+ ;; Needs python-pyspark.
+ "hyperopt/tests/test_spark.py"
+ ;; Needs both python-scikit-learn and python-lightgbm.
+ "--ignore" "hyperopt/tests/test_atpe_basic.py"
+ ;; The tests below need python-lightgbm.
+ "-k" (string-append "not test_branin"
+ " and not test_distractor"
+ " and not test_q1lognormal"
+ " and not test_quadratic1"
+ " and not test_twoarms"))))))))
+ (propagated-inputs
+ `(("python-cloudpickle" ,python-cloudpickle)
+ ("python-future" ,python-future)
+ ("python-networkx" ,python-networkx)
+ ("python-numpy" ,python-numpy)
+ ("python-scipy" ,python-scipy)
+ ("python-six" ,python-six)
+ ("python-tqdm" ,python-tqdm)))
+ (native-inputs
+ `(("python-black" ,python-black)
+ ("python-ipython" ,python-ipython)
+ ("python-ipyparallel" ,python-ipyparallel)
+ ("python-nose" ,python-nose)
+ ("python-pymongo" ,python-pymongo)
+ ("python-pytest" ,python-pytest)))
+ (home-page "https://hyperopt.github.io/hyperopt/")
+ (synopsis "Library for hyperparameter optimization")
+ (description "Hyperopt is a Python library for serial and parallel
+optimization over awkward search spaces, which may include real-valued,
+discrete, and conditional dimensions.")
+ (license license:bsd-3)))
+
;; There have been no proper releases yet.
(define-public kaldi
(let ((commit "d4791c0f3fc1a09c042dac365e120899ee2ad21e")