tfp.experimental.lazybones.utils.plot_graph

Plots leaves and ancestors. (See help(nx.draw_networkx) for kwargs).

tfp.experimental.lazybones.utils.plot_graph(
    leaves, pos=None, with_labels=True, arrowsize=10, node_size=1200, fig=None,
    labels=(lambda node: node.name), seed=42, **kwargs
)