BERTopic 使用零样本主题名称

6kkfgxo0  于 3个月前  发布在  其他
关注(0)|答案(1)|浏览(44)

在模型输出中,是否有办法使用零样本主题名称而不是表示?例如,当我使用zeroshot_min_similarity = 0进行零样本建模时,我只得到零样本主题,但当我调用topic_model.view_hierarchy()时,它显示的是主题关键字表示而不是实际的零样本名称。

mzillmmw

mzillmmw1#

你检查过.visualize_hierarchy的参数了吗?有一个叫做custom_labels的参数,允许你使用自定义标签。要设置自定义标签,请参考这里的文档。只需使用.set_topic_labels函数来设置任何你想要的标签。因此,你可以使用该函数来设置零样本主题,然后调用custom_labels=True在整个可视化中使用它们。

相关问题