diff options
Diffstat (limited to 'pkgs/development/python-modules/imbalanced-learn/0.4.nix')
-rw-r--r-- | pkgs/development/python-modules/imbalanced-learn/0.4.nix | 29 |
1 files changed, 29 insertions, 0 deletions
diff --git a/pkgs/development/python-modules/imbalanced-learn/0.4.nix b/pkgs/development/python-modules/imbalanced-learn/0.4.nix new file mode 100644 index 00000000000..e7d2c2f3714 --- /dev/null +++ b/pkgs/development/python-modules/imbalanced-learn/0.4.nix @@ -0,0 +1,29 @@ +{ stdenv, buildPythonPackage, fetchPypi, scikitlearn, pandas, nose, pytest }: + +buildPythonPackage rec { + pname = "imbalanced-learn"; + version = "0.4.3"; + + src = fetchPypi { + inherit pname version; + sha256 = "5bd9e86e40ce4001a57426541d7c79b18143cbd181e3330c1a3e5c5c43287083"; + }; + + propagatedBuildInputs = [ scikitlearn ]; + checkInputs = [ nose pytest pandas ]; + checkPhase = '' + export HOME=$PWD + # skip some tests that fail because of minimal rounding errors + # or large dependencies + py.test imblearn -k 'not classification \ + and not _generator \ + and not _forest \ + and not wrong_memory' + ''; + + meta = with stdenv.lib; { + description = "Library offering a number of re-sampling techniques commonly used in datasets showing strong between-class imbalance"; + homepage = https://github.com/scikit-learn-contrib/imbalanced-learn; + license = licenses.mit; + }; +} |