diff --git a/train.py b/train.py
index 236750b1..a20da118 100644
--- a/train.py
+++ b/train.py
@@ -13,7 +13,7 @@
 
 # Fit a model
 
-depth = 5
+depth = 3
 
 clf = RandomForestClassifier(max_depth=depth)
 clf.fit(X_train, y_train)