From 0641835390c9d22ca90df5abe16ae747583a3876 Mon Sep 17 00:00:00 2001 From: Adriane Boyd Date: Wed, 2 Aug 2023 13:39:29 +0200 Subject: [PATCH] Add hashCount and mode to python train_supervised --- python/floret_module/floret/floret.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/python/floret_module/floret/floret.py b/python/floret_module/floret/floret.py index 6dcd46b40..6212ab604 100644 --- 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)