python - 优化函数评估缓存部分结果

Question

假设我有一个带有许多输入参数的复杂数学函数P = [p1, ..., pn]。假设我可以将函数分解为块，例如：

f(P) = f1(p1, p2) * f2(p2, ... pn)

有可能

f2(p2, ..., pn) = p2 * f3(p4) + f4(p5, ..., pn)

假设我必须评估f的许多值P，例如我想找到 的最小值f。假设我已经计算f(P)并且我需要计算f(P')where P'is equal to Pexcept p1。在这种情况下，我不必重新计算f2, f3, f4，而只需f1.

是否有一个库可以帮助我实现这种缓存系统？我知道RooFit，但它面向统计模型，由块组成。我正在寻找更一般的东西。首选 scipy / scikits 和类似的，但 c++ 库也可以。这种技术有名字吗？

Answer 1

如果你可以将这些函数写成纯函数（这意味着它们总是为相同的参数返回相同的值，并且没有副作用），你可以使用 memoization，这是一种保存函数调用结果的方法。

try:
    from functools import lru_cache  # Python 3.2+
except ImportError:  # Python 2
    # Python 2 and Python 3.0-3.1
    # Requires the third-party functools32 package
    from functools32 import lru_cache

@lru_cache(maxsize=None)
def f(arg):
    # expensive operations
    return result

x = f('arg that will take 10 seconds')  # takes 10 seconds
y = f('arg that will take 10 seconds')  # virtually instant

为了说明，或者如果您不想functools32在 Python < 3.2 上使用：

def memoize(func):
    memo = {}

    def wrapper(*args):
        if args not in memo:            
            memo[args] = func(*args)
        return memo[args]

    return helper

@memoize
def f(arg):
    # expensive operations
    return result

x = f('arg that will take 10 seconds')  # takes 10 seconds
y = f('arg that will take 10 seconds')  # virtually instant