diff --git a/src/main/cpp/fastText b/src/main/cpp/fastText index 9c9b9b2..501b9b1 160000 --- a/src/main/cpp/fastText +++ b/src/main/cpp/fastText @@ -1 +1 @@ -Subproject commit 9c9b9b2eca4fdc0182182cb69234fffbb847d820 +Subproject commit 501b9b1e4543fd2de55e4a621a9924ce7d2b5b17 diff --git a/src/main/cpp/fasttext_wrapper.cc b/src/main/cpp/fasttext_wrapper.cc index ec5c90e..94eec05 100644 --- a/src/main/cpp/fasttext_wrapper.cc +++ b/src/main/cpp/fasttext_wrapper.cc @@ -74,7 +74,11 @@ namespace FastTextWrapper { const std::string& text, int32_t k) { std::vector> predictions; std::istringstream in(text); - fastText.predict(in, k, predictions); + + // temporary workaround + real th = 0.0; + + fastText.predictLine(in, predictions, k, th); return predictions; }