diff options
Diffstat (limited to 'requirements.txt')
| -rw-r--r-- | requirements.txt | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/requirements.txt b/requirements.txt index 765977f..7cf8570 100644 --- a/requirements.txt +++ b/requirements.txt @@ -2,4 +2,5 @@ torch~=1.7.1 torchvision~=0.8.0a0+ecf4e9c numpy~=1.19.4 tqdm~=4.55.0 -Pillow~=8.0.1
\ No newline at end of file +Pillow~=8.0.1 +scikit-learn~=0.23.2
\ No newline at end of file |