在模型输出中,是否有办法使用零样本主题名称而不是表示?例如,当我使用zeroshot_min_similarity = 0进行零样本建模时,我只得到零样本主题,但当我调用topic_model.view_hierarchy()时,它显示的是主题关键字表示而不是实际的零样本名称。
mzillmmw1#
你检查过.visualize_hierarchy的参数了吗?有一个叫做custom_labels的参数,允许你使用自定义标签。要设置自定义标签,请参考这里的文档。只需使用.set_topic_labels函数来设置任何你想要的标签。因此,你可以使用该函数来设置零样本主题,然后调用custom_labels=True在整个可视化中使用它们。
.visualize_hierarchy
custom_labels
.set_topic_labels
custom_labels=True
1条答案
按热度按时间mzillmmw1#
你检查过
.visualize_hierarchy
的参数了吗?有一个叫做custom_labels
的参数,允许你使用自定义标签。要设置自定义标签,请参考这里的文档。只需使用.set_topic_labels
函数来设置任何你想要的标签。因此,你可以使用该函数来设置零样本主题,然后调用custom_labels=True
在整个可视化中使用它们。