diff --git a/requirements.txt b/requirements.txt index 1450efd..a790bd1 100644 --- a/requirements.txt +++ b/requirements.txt @@ -2,6 +2,7 @@ Cython==0.29.21 librosa==0.9.2 numpy==1.21.6 scipy==1.5.2 +scikit-learn==1.0.2 tensorboard torch==1.13.1 torchvision==0.14.1