Переглянути джерело

autotune best train arguments reported to python model object

Summary:
following issues #887 and #913: Print out the best parameters from autotune
This diff makes it possible to access the best train parameters found by autotune in the python model object.

Reviewed By: EdouardGrave

Differential Revision: D19142951

fbshipit-source-id: 7a95a234652530358d398fe06229d0c67bd8674f
Onur Çelebi 6 роки тому
батько
коміт
5e1320a159
1 змінених файлів з 4 додано та 0 видалено
  1. 4 0
      python/fasttext_module/fasttext/FastText.py

+ 4 - 0
python/fasttext_module/fasttext/FastText.py

@@ -43,7 +43,9 @@ class _FastText(object):
             self.f.loadModel(model_path)
         self._words = None
         self._labels = None
+        self.set_args(args)
 
+    def set_args(self, args=None):
         if args:
             arg_names = ['lr', 'dim', 'ws', 'epoch', 'minCount',
                          'minCountLabel', 'minn', 'maxn', 'neg', 'wordNgrams',
@@ -453,6 +455,7 @@ def train_supervised(*kargs, **kwargs):
     a = _build_args(args, manually_set_args)
     ft = _FastText(args=a)
     fasttext.train(ft.f, a)
+    ft.set_args(ft.f.getArgs())
     return ft
 
 
@@ -478,6 +481,7 @@ def train_unsupervised(*kargs, **kwargs):
     a = _build_args(args, manually_set_args)
     ft = _FastText(args=a)
     fasttext.train(ft.f, a)
+    ft.set_args(ft.f.getArgs())
     return ft