diff --git a/requirements.txt b/requirements.txt
index 696f6a5..309f47c 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -4,7 +4,7 @@ pandas==0.22.0
 scipy==1.0.1
 scikit-learn==0.19.1
 six==1.11.0
-PyYAML==3.12
+PyYAML==5.4
 wheel==0.31.0
 requests==2.18.4
 Pillow>=1.0