tjh.dev
1{
2 lib,
3 buildPythonPackage,
4 fetchFromGitHub,
5 pythonOlder,
6 pytestCheckHook,
7 bokeh,
8 ipython,
9 matplotlib,
10 numpy,
11 nbconvert,
12 nbformat,
13}:
14
15buildPythonPackage rec {
16 pname = "livelossplot";
17 version = "0.5.5";
18 format = "setuptools";
19
20 disabled = pythonOlder "3.6";
21
22 src = fetchFromGitHub {
23 owner = "stared";
24 repo = pname;
25 rev = "v${version}";
26 sha256 = "sha256-YU8vX4SubI6txmC/i5fOjcvWfuDFm8+SPmie8Eb1qRs=";
27 };
28
29 propagatedBuildInputs = [
30 bokeh
31 ipython
32 matplotlib
33 numpy
34 ];
35
36 nativeCheckInputs = [
37 nbconvert
38 nbformat
39 pytestCheckHook
40 ];
41
42 meta = with lib; {
43 description = "Live training loss plot in Jupyter for Keras, PyTorch, and others";
44 homepage = "https://github.com/stared/livelossplot";
45 license = licenses.mit;
46 maintainers = with maintainers; [ bcdarwin ];
47 };
48}