Update train.py

iterative · Jan 30, 2024 · 53c55db · 53c55db
1 parent 65512b6
commit 53c55db
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/train.py b/train.py
@@ -12,13 +12,9 @@
 y_test = np.genfromtxt("data/test_labels.csv")
 
 # Fit a model
-<<<<<<< HEAD
 
-depth = 4
-
-=======
 depth = 5
->>>>>>> c9c93716fd5af55e194fccb17839ace3add989cd
+
 clf = RandomForestClassifier(max_depth=depth)
 clf.fit(X_train, y_train)