Commit 0f25da38 authored by Elle O'Brien's avatar Elle O'Brien
Browse files

Update train.py

parent c3e495ca
......@@ -13,7 +13,7 @@ y_test = np.genfromtxt("data/test_labels.csv")
# Fit a model
depth = 10
depth = 5
clf = RandomForestClassifier(max_depth=depth)
clf.fit(X_train,y_train)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment