From d248ef0e0c9c9e7b924c3508f43e339687975627 Mon Sep 17 00:00:00 2001 From: Vasil Zlatanov Date: Mon, 4 Feb 2019 19:02:43 +0000 Subject: Remove hardcoded leaf limit --- evaluate.py | 4 ++-- 1 file 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") -- cgit v1.2.3-54-g00ecf