Pārlūkot izejas kodu

getDictionary api for FastText

Summary: Enable users to get const pointer to Dictionary.

Reviewed By: stephenroller, EdouardGrave

Differential Revision: D5826941

fbshipit-source-id: 66442d7c6428c79aa74fbbce270aba3feccd34fa
Christian Puhrsch 8 gadi atpakaļ
vecāks
revīzija
7e5c955a36
2 mainītis faili ar 5 papildinājumiem un 0 dzēšanām
  1. 4 0
      src/fasttext.cc
  2. 1 0
      src/fasttext.h

+ 4 - 0
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;

+ 1 - 0
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();