Add hashCount and mode to python train_supervised (#27)

explosion · Aug 2, 2023 · 82d089b · 82d089b
1 parent 6f78c26
commit 82d089b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python/floret_module/floret/floret.py b/python/floret_module/floret/floret.py
@@ -544,7 +544,8 @@ def train_supervised(*kargs, **kwargs):
                  'minCountLabel', 'minn', 'maxn', 'neg', 'wordNgrams', 'loss', 'bucket',
                  'thread', 'lrUpdateRate', 't', 'label', 'verbose', 'pretrainedVectors',
                  'seed', 'autotuneValidationFile', 'autotuneMetric',
-                 'autotunePredictions', 'autotuneDuration', 'autotuneModelSize']
+                 'autotunePredictions', 'autotuneDuration', 'autotuneModelSize',
+                 'hashCount', 'mode']
     args, manually_set_args = read_args(kargs, kwargs, arg_names,
                                         supervised_default)
     a = _build_args(args, manually_set_args)