about summary refs log tree commit diff
diff options
context:
space:
mode:
-rw-r--r--pkgs/development/python-modules/sklearn-deap/default.nix29
-rw-r--r--pkgs/top-level/python-packages.nix2
2 files changed, 31 insertions, 0 deletions
diff --git a/pkgs/development/python-modules/sklearn-deap/default.nix b/pkgs/development/python-modules/sklearn-deap/default.nix
new file mode 100644
index 000000000000..2c40ce4c2fc3
--- /dev/null
+++ b/pkgs/development/python-modules/sklearn-deap/default.nix
@@ -0,0 +1,29 @@
+{ stdenv, buildPythonPackage, fetchFromGitHub, numpy, scipy, deap, scikitlearn, python }:
+
+buildPythonPackage rec {
+  pname = "sklearn-deap";
+  version = "0.2.2";
+
+  # No tests in Pypi
+  src = fetchFromGitHub {
+    owner = "rsteca";
+    repo = pname;
+    rev = "${version}";
+    sha256 = "01ynmzxg181xhv2d7bs53zjvk9x2qpxix32sspq54mpigxh13ava";
+  };
+
+  propagatedBuildInputs = [ numpy scipy deap scikitlearn ];
+
+  #doCheck = false;
+  checkPhase = ''
+    ${python.interpreter} test.py
+  '';
+
+  meta = with stdenv.lib; {
+    description = "Use evolutionary algorithms instead of gridsearch in scikit-learn";
+    homepage = https://github.com/rsteca/sklearn-deap;
+    license = licenses.lgpl3;
+    maintainers = with maintainers; [ psyanticy ];
+  };
+}
+
diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix
index d057cebf9f57..83af2a4a3107 100644
--- a/pkgs/top-level/python-packages.nix
+++ b/pkgs/top-level/python-packages.nix
@@ -464,6 +464,8 @@ in {
 
   sip = callPackage ../development/python-modules/sip { };
 
+  sklearn-deap = callPackage ../development/python-modules/sklearn-deap { };
+
   spglib = callPackage ../development/python-modules/spglib { };
 
   supervise_api = callPackage ../development/python-modules/supervise_api { };