diff --git a/requirements.txt b/requirements.txt
index 5e223a0..198bee2 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -11,4 +11,4 @@ ipython==6.1.0
 Pillow==4.2.1
 seaborn==0.7.1
 tensorflow==1.2.1
-PyYAML==3.12
+PyYAML==5.4