File size: 522 Bytes
15a9463 5aa4afe 488f722 5aa4afe 7cc0ebc |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
---
license: mit
---
# BERTopicを日本語webデータで訓練したモデルです。
- modelはtopic_model.binを直接wgetなどでダウンロードください。
~~~
from bertopic import BERTopic
model_path = "data/topic_model.bin"
topic_model = BERTopic.load(model_path)
docs=["こんにちは",
"吾輩は猫である",
]
categories = topic_model.transform(docs)[0]
~~~
- 訓練コードはtrain_topic_model.ipynbを参照ください
- 340カテゴリ程度に分割されます
- 履歴
- 2024/3/12訓練 |