diff --git a/requirements.txt b/requirements.txt index f6c4b66..562fed3 100644 --- a/requirements.txt +++ b/requirements.txt @@ -2,5 +2,5 @@ torch==1.5.0 torchtext six==1.11.0 pytorch-crf==0.5.1 -scikit-learn==0.19.1 +scikit-learn==1.5.0 pytorch-lightning==0.7.6 \ No newline at end of file