diff options
| -rwxr-xr-x | evaluate.py | 4 |
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") |