fewer elements in grid search

dedupeio · Jun 2, 2022 · 574aa84 · 574aa84
1 parent c84a0c8
commit 574aa84
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dedupe/api.py b/dedupe/api.py
@@ -1092,7 +1092,7 @@ def __init__(
         ]
         self.classifier = sklearn.model_selection.GridSearchCV(
             estimator=sklearn.ensemble.RandomForestClassifier(),
-            param_grid={"n_estimators": [100, 200, 400, 800, 1600]},
+            param_grid={"n_estimators": [100, 200, 400, 800]},
             scoring="f1",
             n_jobs=-1,
         )