使用 Python 第三方库 daft 绘制 PGM 中的贝叶斯网络

daft 的官方文档请见 DAFT：BEAUTIFULLY RENDERED PROBABILISTIC GRAPHICAL MODELS.

from matplotlib import rcrc("font", family="serif", size=12)rc("text", usetex=True)import daftpgm = daft.PGM([3.6, 2.7], origin=[1.15, 0.65])pgm.add_node(daft.Node("cloudy", r"cloudy", 3, 3, aspect=1.8))pgm.add_node(daft.Node("rain", r"rain", 2, 2, aspect=1.2))pgm.add_node(daft.Node("sprinkler", r"sprinkler", 4, 2, aspect=2.1))pgm.add_node(daft.Node("wet", r"grass wet", 3, 1, aspect=2.4, observed=True))pgm.add_edge("cloudy", "rain")pgm.add_edge("cloudy", "sprinkler")pgm.add_edge("rain", "wet")pgm.add_edge("sprinkler", "wet")pgm.render()pgm.figure.savefig("wordy.pdf")pgm.figure.savefig("wordy.png", dpi=150)