fasttextでテキスト分類をしようと思っているのですがpredict_probaを使用するとエラーが出てしまいます
pythonでfasttextで簡単なプログラムを実行したのですが、
attributeerrorが出て解決できません。
発生している問題・エラーメッセージ
Traceback (most recent call last): File "classifier.py", line 8, in <module> predict("おはようございます") File "classifier.py", line 5, in predict estimate = classifier.predict_proba([word], k=2) AttributeError: '_FastText' object has no attribute 'predict_proba'
該当のソースコード
python
1import fasttext as ft 2 3def predict(word): 4 classifier = ft.load_model('model.bin') 5 estimate = classifier.predict_proba([word], k=2) 6 print(estimate) 7 8predict("おはようございます")
試したこと
fasttext内で実行してます。
pipなどでインストールしているのは確認しました。
win10のubuntuで実行しています。
インストールしたfasttextはどれでしょうか?(Facebook公式のものですか?)
初心者でよくわからずサイトを色々見てインストールしたので定かではないですが
$ git clone https://github.com/facebookresearch/fastText.git
$ cd fastText
$ make
$ pip3 install fasttext
でインストールしたと思います。
確認したところ「predict()」はあるのですが、「predict_proba()」は見当たりません。
https://fasttext.cc/docs/en/html/classfasttext_1_1FastText.html#ac844ddc1573e80a8b5a255668fc97247
質問のコードは何かサイト等を参考にされましたか? もしかしたら、仕様変更などあったかもしれませんね。
qiitaなどを参考にしました。確かに、ないですね。
一度predict()で試してみます。
predict()で実行できました。ありがとうございます。
回答1件
あなたの回答
tips
プレビュー