diff --git a/train.py b/train.py
index ba4c71e7e..d3ed33a14 100644
--- a/train.py
+++ b/train.py
@@ -13,7 +13,7 @@
 
 # Fit a model
 
-depth = 5
+depth = 3
 
 
 clf = RandomForestClassifier(max_depth=depth)