mirrors / cleanlab--cleanlab

项目文件夹

文件

T

Jonas Mueller b93fdebf30 Add compatibility for tensorflow and pytorch Dataset objects (#311 )

* validation_func docstring

* torch,tf compatibility+tests

* keras test

* skip tests if python < 3.7

* pytorch numpy int bug on windows

* make tensorflow test work on windows

* move tf env variable setting

* pytorch test increase epochs

* install cpu-tensorflow on windows CI

* torch test optimizer to adam

* fix bugs in shuffled TF dataset

* dummy unit test for TF on windows

* dummy code for TF windows testing

* deal with np.int bug on windows

* remove windows debugging code

* docstrings for new functionality

* address merge conflicts

* reformat after merge

* addressed comments

2022-07-27 21:42:13 -07:00

13 行

161 B

Plaintext

原始文件 永久链接 Blame 文件历史

 # Python dependencies for development
 coverage != 6.3, != 6.3.*
 mypy
 pandas-stubs
 pre-commit
 pytest
 pytest-cov
 requests
 scipy
 torch
 torchvision
 skorch
 tensorflow