python3Packages.skl2onnx: init at 1.13
This commit is contained in:
parent
18f7d4c992
commit
d5f27c97b8
2 changed files with 51 additions and 0 deletions
49
pkgs/development/python-modules/skl2onnx/default.nix
Normal file
49
pkgs/development/python-modules/skl2onnx/default.nix
Normal file
|
@ -0,0 +1,49 @@
|
|||
{ lib
|
||||
, buildPythonPackage
|
||||
, fetchPypi
|
||||
, numpy
|
||||
, scipy
|
||||
, protobuf
|
||||
, onnx
|
||||
, scikit-learn
|
||||
, onnxconverter-common
|
||||
, onnxruntime
|
||||
, pandas
|
||||
, unittestCheckHook
|
||||
}:
|
||||
|
||||
buildPythonPackage rec {
|
||||
pname = "skl2onnx";
|
||||
version = "1.13";
|
||||
|
||||
src = fetchPypi {
|
||||
inherit pname version;
|
||||
hash = "sha256-XzUva5uFX/rGMFpwfwLH1Db0Nok47pBJCSqVo1ZcJz0=";
|
||||
};
|
||||
|
||||
propagatedBuildInputs = [
|
||||
numpy
|
||||
scipy
|
||||
protobuf
|
||||
onnx
|
||||
scikit-learn
|
||||
onnxconverter-common
|
||||
];
|
||||
|
||||
checkInputs = [
|
||||
onnxruntime
|
||||
pandas
|
||||
unittestCheckHook
|
||||
];
|
||||
|
||||
unittestFlagsArray = [ "-s" "tests" ];
|
||||
|
||||
# Core dump
|
||||
doCheck = false;
|
||||
|
||||
meta = {
|
||||
description = "Convert scikit-learn models to ONNX";
|
||||
maintainers = with lib.maintainers; [ fridh ];
|
||||
license = with lib.licenses; [ asl20 ];
|
||||
};
|
||||
}
|
|
@ -10167,6 +10167,8 @@ in {
|
|||
|
||||
skidl = callPackage ../development/python-modules/skidl { };
|
||||
|
||||
skl2onnx = callPackage ../development/python-modules/skl2onnx { };
|
||||
|
||||
sklearn-deap = callPackage ../development/python-modules/sklearn-deap { };
|
||||
|
||||
skodaconnect = callPackage ../development/python-modules/skodaconnect { };
|
||||
|
|
Loading…
Reference in a new issue