parent
4303b0a517
commit
3f50332bc4
@ -0,0 +1,70 @@ |
||||
{ stdenv |
||||
, buildPythonPackage |
||||
, fetchFromGitHub |
||||
, isPy27 |
||||
, pytestCheckHook |
||||
, numpy |
||||
, scipy |
||||
, scikitlearn |
||||
, pandas |
||||
, tqdm |
||||
, slicer |
||||
, numba |
||||
, matplotlib |
||||
, nose |
||||
, ipython |
||||
}: |
||||
|
||||
buildPythonPackage rec { |
||||
pname = "shap"; |
||||
version = "0.36.0"; |
||||
disabled = isPy27; |
||||
|
||||
src = fetchFromGitHub { |
||||
owner = "slundberg"; |
||||
repo = pname; |
||||
rev = "v${version}"; |
||||
sha256 = "1wxnxvbz6avzzfqjfbcqd4v879hvpq4021v31fhdpccr2q317rr9"; |
||||
}; |
||||
|
||||
propagatedBuildInputs = [ |
||||
numpy |
||||
scipy |
||||
scikitlearn |
||||
pandas |
||||
tqdm |
||||
slicer |
||||
numba |
||||
]; |
||||
|
||||
preCheck = '' |
||||
export HOME=$TMPDIR |
||||
# when importing the local copy the extension is not found |
||||
rm -r shap |
||||
''; |
||||
checkInputs = [ pytestCheckHook matplotlib nose ipython ]; |
||||
# Those tests access the network |
||||
disabledTests = [ |
||||
"test_kernel_shap_with_a1a_sparse_zero_background" |
||||
"test_kernel_shap_with_a1a_sparse_nonzero_background" |
||||
"test_kernel_shap_with_high_dim_sparse" |
||||
"test_sklearn_random_forest_newsgroups" |
||||
"test_sum_match_random_forest" |
||||
"test_sum_match_extra_trees" |
||||
"test_single_row_random_forest" |
||||
"test_sum_match_gradient_boosting_classifier" |
||||
"test_single_row_gradient_boosting_classifier" |
||||
"test_HistGradientBoostingClassifier_proba" |
||||
"test_HistGradientBoostingClassifier_multidim" |
||||
"test_sum_match_gradient_boosting_regressor" |
||||
"test_single_row_gradient_boosting_regressor" |
||||
]; |
||||
|
||||
meta = with stdenv.lib; { |
||||
description = "A unified approach to explain the output of any machine learning model"; |
||||
homepage = "https://github.com/slundberg/shap"; |
||||
license = licenses.mit; |
||||
maintainers = with maintainers; [ evax ]; |
||||
platforms = platforms.unix; |
||||
}; |
||||
} |
@ -0,0 +1,29 @@ |
||||
{ stdenv |
||||
, buildPythonPackage |
||||
, fetchPypi |
||||
, isPy27 |
||||
, pytestCheckHook |
||||
, pandas |
||||
, pytorch |
||||
}: |
||||
|
||||
buildPythonPackage rec { |
||||
pname = "slicer"; |
||||
version = "0.0.4"; |
||||
disabled = isPy27; |
||||
|
||||
src = fetchPypi { |
||||
inherit pname version; |
||||
sha256 = "21d53aac4e78c93fd83c0fd2f8f9d8a2195ac079dffdc0da81cd749da0f2f355"; |
||||
}; |
||||
|
||||
checkInputs = [ pytestCheckHook pandas pytorch ]; |
||||
|
||||
meta = with stdenv.lib; { |
||||
description = "Wraps tensor-like objects and provides a uniform slicing interface via __getitem__"; |
||||
homepage = "https://github.com/interpretml/slicer"; |
||||
license = licenses.mit; |
||||
maintainers = with maintainers; [ evax ]; |
||||
platforms = platforms.unix; |
||||
}; |
||||
} |
Loading…
Reference in new issue