getDictionary api for FastText

Summary: Enable users to get const pointer to Dictionary. Reviewed By: stephenroller, EdouardGrave Differential Revision: D5826941 fbshipit-source-id: 66442d7c6428c79aa74fbbce270aba3feccd34fa
jattenberg · Sep 16, 2017 · 7e5c955 · 7e5c955
1 parent 1f69d86
commit 7e5c955
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/src/fasttext.cc b/src/fasttext.cc
@@ -399,6 +399,10 @@ void FastText::sentenceVectors() {
   }
 }
 
+std::shared_ptr<const Dictionary> FastText::getDictionary() const {
+  return dict_;
+}
+
 void FastText::ngramVectors(std::string word) {
   std::vector<int32_t> ngrams;
   std::vector<std::string> substrings;

diff --git a/src/fasttext.h b/src/fasttext.h
@@ -54,6 +54,7 @@ class FastText {
     FastText();
 
     void getVector(Vector&, const std::string&) const;
+    std::shared_ptr<const Dictionary> getDictionary() const;
     void saveVectors();
     void saveOutput();
     void saveModel();