diff --git a/requirements.txt b/requirements.txt
index 64dcdf7..86f6dc9 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -2,6 +2,7 @@ torch>=1.2
 hydra-core>=0.11
 tensorboard>=2.0
 matplotlib>=3.1
-transformers>=2.0
+scikit-learn>=0.22
+transformers>=2.2
 jieba>=0.39
 pyhanlp>=0.1.57
\ No newline at end of file