我正在关注 this tutorial并创建了一个这样的图表:
from dask.threaded import get
from operator import add
dsk = {
'x': 1,
'y': 2,
'z': (add, 'x', 'y'),
'w': (sum, ['x', 'y', 'z'])
}
get(dsk, "w")
成功了,我得到了想要的输出。如何可视化计算图? visualize 方法需要一个 DASK 对象,而我只有一个字典。
提前致谢!
最佳答案
dask.visualize
适用于 Dask Collections -- API docs here提到 args
需要是一个“dask 对象”,这意味着一个 Dask 集合(我已经打开 this issue 来改进文档!)。
因此,如果您将任务图 dsk
包装在一个集合中,您应该能够可视化
它:
import dask
from operator import add
from dask.threaded import get
from dask.delayed import Delayed
dsk = {
'x': 1,
'y': 2,
'z': (add, 'x', 'y'),
'w': (sum, ['x', 'y', 'z'])
}
# wrapping dsk in a Dask Collection (Delayed)
delayed_dsk = Delayed("w", dsk)
# call visualize as usual
delayed_dsk.visualize()
# Or,
dask.visualize(delayed_dsk)
关于python - 可视化 DASK 任务图,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/70880012/