diff --git a/requirements.txt b/requirements.txt
index 8ca0a42..82c1802 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -4,4 +4,4 @@ tqdm==4.11.2
 matplotlib==1.5.3
 scikit_learn==0.19.1
 scipy==1.0.0
-pyyaml==5.1.2
+pyyaml==5.4