1 files changed, 2 insertions, 2 deletions
diff --git a/evaluate.py b/evaluate.py
index 9e3a613..92c4107 100755
--- a/evaluate.py
+++ b/evaluate.py
@@ -7,7 +7,7 @@ CLUSTER_CNT = 256
 KMEANS = False
 
 if KMEANS:
-    N_ESTIMATORS = 10
+    N_ESTIMATORS = 1000
 else:
     N_ESTIMATORS = 1
 
@@ -29,7 +29,7 @@ if (KMEANS):
     print("Computing KMeans with", train_part.shape[0], "keywords")
     kmeans = KMeans(n_clusters=CLUSTER_CNT, n_init=N_ESTIMATORS, random_state=0).fit(train_part)
 else:
-    trees = RandomTreesEmbedding(max_leaf_nodes=256, n_estimators=N_ESTIMATORS, random_state=0).fit(train_part)
+    trees = RandomTreesEmbedding(max_leaf_nodes=CLUSTER_CNT, n_estimators=N_ESTIMATORS, random_state=0).fit(train_part)
 
 
 print("Generating histograms")