BERTopic 创建带有自定义标签的主题树

xxhby3vn  于 22天前  发布在  其他
关注(0)|答案(5)|浏览(28)

你好,我正在尝试打印主题树以获得基于层次结构的文本可视化。有没有办法用表示模型中的CustomName替换hierarchical_topics()中的Parent_Name、Child_Left_Name和Child_Right_Name值?

vm0i2vca

vm0i2vca1#

目前这是不可能的,因为自定义标签仅针对在.fit.fit_transform期间找到的主题创建,而不是使用.hierarchical_topics找到的主题。如果有足够的兴趣,这可以包含在未来的版本中。

frebpwbc

frebpwbc2#

好的,这肯定是一个有趣的特点,至少对我来说是这样!现在我正在使用llama2作为主题表示模型,所以可能可以将这个树反馈回它,让它为父节点生成标签。感谢回复!

z9ju0rcb

z9ju0rcb3#

这确实是一个不错的解决方案!在多层层次结构和多种方法的情况下,计算多个方面可能会变得棘手,但我一定会确保跟踪这个问题。

slhcrj9b

slhcrj9b4#

目前这是不可能的,因为自定义标签仅针对在.fit.fit_transform期间找到的主题创建,而不是使用.hierarchical_topics找到的主题。如果有足够的兴趣,这可以包含在未来的版本中。
非常感兴趣!

相关问题