Dask混淆矩阵

Question

我正在尝试使用Dask计算混淆矩阵元素。 从算法的角度来看，我的实现似乎还可以。 但是，当我在大小分别为100万的2个阵列上运行它时，这将永远花费。

有人对如何优化此代码有建议吗？

1s

Answer 1

您应该意识到的一项快速改进：

import dask
TP, FP, TN, FN = dask.compute(TP, FP, TN, FN)

而不是分别调用.compute()。这样可以共享通用数据和任务，从而减少了要完成的工作总量。