python - Dask 在 Dask 图中可视化多个输出节点

Question

我正在创建的 Dask 图有多个输出。

我想知道是否可以同时可视化多个 dask 输出。

当我尝试使用dask.visualize(graph). Daskgraph节点的元组或字典在哪里。它似乎因生成空图像而失败。

例如：

import dask

@dask.delayed
def op(x, y):
    return x + y

def graph():
    a = 1
    b = 2
    c = 3
    d = 4

    f = op(a, b)
    g = op(c, d)
    h = op(a, d)

    output1 = op(f, g)
    output2 = op(g, h)

    # works
    dask.visualize(output1, filename='output1.png')
    dask.visualize(output2, filename='output2.png')

    # empty small square box
    dask.visualize((output1, output2), filename='output1and2.png')
    # only outputs the name of the keys
    dask.visualize({'output1': output1, 'output2': output2}, filename='output1and2_.png')

graph()

Answer 1

该dask.visualize函数接受许多参数，而不是一系列参数

做这个

dask.visualize(output1, output2, filename='output1and2.png')

而不是这个

dask.visualize((output1, output2), filename='output1and2.png')