需要手动将0th topic中的数据添加到visualization数据中,并将迭代指针调整为1。 以下是代码样例:
from bertopic import BERTopic
from bertopic import visualisation
# 创建BERTopic模型
model = BERTopic()
# 在数据上拟合模型
topics, _ = model.fit_transform(docs)
# 将0th topic中的数据添加到visualization数据中
visual_data = model.visualize_topics_over_time(docs, topics, topics_over_time)
# 将迭代指针调整为1
iterator = iter(visual_data)
next(iterator)
visualisation.visualize_topics_over_time(visual_data, top_n_topics=5)