更改使用黄砖创建的分散图中的标题

Question

我想更改使用 Yellowbrick 创建的散布图的标题：

我使用以下代码：

wl = []
with open('my-text.txt', 'r', encoding='utf8') as f:
   wl = f.read().split()
   
topics = ['δὲ']
plt.figure(figsize=(16,.5))
visualizer = DispersionPlot(topics)
visualizer.fit([poll_wl])
visualizer.show()

创建的情节的标题是“词汇散布情节”。我想使用自定义标题。

Answer 1

您可以从 Visalizer 中获取 figure，然后像这样设置唯一 axes 对象的标题文本：

visualizer.fig.get_axes()[0].title.set_text("My Title")

visualizer.set_title("My Title") 也可以这样做，尽管它似乎没有明确包含在 docs 中。

示例（来自the docs）：

from yellowbrick.text import DispersionPlot
from yellowbrick.datasets import load_hobbies

corpus = load_hobbies()
text = [doc.split() for doc in corpus.data]
target_words = ['Game', 'player', 'score', 'oil', 'Man']
visualizer = DispersionPlot(target_words)
visualizer.fit(text)
visualizer.show()

visualizer.set_title("My Title")