tjh.dev
1{ stdenv, buildPythonPackage, fetchPypi, scikitlearn, pandas, nose, pytest }:
2
3buildPythonPackage rec {
4 pname = "imbalanced-learn";
5 version = "0.4.3";
6
7 src = fetchPypi {
8 inherit pname version;
9 sha256 = "5bd9e86e40ce4001a57426541d7c79b18143cbd181e3330c1a3e5c5c43287083";
10 };
11
12 propagatedBuildInputs = [ scikitlearn ];
13 checkInputs = [ nose pytest pandas ];
14 checkPhase = ''
15 export HOME=$PWD
16 # skip some tests that fail because of minimal rounding errors
17 py.test imblearn --ignore=imblearn/metrics/classification.py
18 py.test doc/*.rst
19 '';
20
21 meta = with stdenv.lib; {
22 description = "Library offering a number of re-sampling techniques commonly used in datasets showing strong between-class imbalance";
23 homepage = https://github.com/scikit-learn-contrib/imbalanced-learn;
24 license = licenses.mit;
25 };
26}