bab2min / tomotopy

Python package of Tomoto, the Topic Modeling Tool
https://bab2min.github.io/tomotopy
MIT License
557 stars 62 forks source link

New optional argument for selecting log-likelihood #149

Open bab2min opened 2 years ago

bab2min commented 2 years ago

Suggested by #147.

result_corpus, ll_per_doc = mdl.infer(corpus)

Currently, ll_per_doc are actually sums of ll for doc-topic and ll for topic-term in above code. But depending on the situation, it may be convenient to return a different ll value. So suggest following a new argument:

result_corpus, ll_per_doc = mdl.infer(corpus, ll=ll_type)

A valid ll_type is one of followings: