保留自定义标签的BERTopic.merge_models()方法,

disho6za  于 4个月前  发布在  其他
关注(0)|答案(4)|浏览(76)

嘿,@MaartenGr ,当合并两个独立的主题模型时,是否可以保留自定义标签?我想能够识别在合并后的模型中保留的特定主题,这些主题来自模型1和模型2。谢谢!

qxgroojn

qxgroojn1#

您是指自定义标签完全没有保存,还是指合并主题中的自定义标签没有保存?

camsedfj

camsedfj2#

自定义标签根本没有保存。
举个例子,这是我测试中的一段代码片段和结果的一部分:

topic_model_jt.set_topic_labels({31: "jtnp_31_aesthetic appreciation", 27: "jtnp_27_wildlife protection"})
topic_model_jt.get_topic_info(31)
merged_model = BERTopic.merge_models([topic_model_base, topic_model_jt], min_similarity =0.99)
merged_topics = merged_model.get_topic_info()
merged_topics.to_csv('test_merged_topics.csv',index=False)

这里是结果:

正如你所见,合并后的模型中没有保存自定义标签。

33qvvth1

33qvvth13#

感谢您的分享!确实看起来它们没有被保存。我需要深入研究一下,看看在将自定义标签与其他主题合并时,我们如何才能最好地保存它们。我有一些代码可以做到这一点,但还有一些细节需要解决。

o2gm4chl

o2gm4chl4#

感谢您的回复。好的,我明白了。如果能及时告知我更新情况,我会非常感激。提前谢谢!:-)

相关问题