创建在原始函数中设置关键字参数的分部函数



不确定这是否可能,但我正在尝试从具有位置和关键字参数的函数创建一个部分函数。 问题是,我希望生成的分部函数中的参数在原始函数中设置关键字参数之一 - 而不是它的位置参数之一。

以下是原始函数的定义:

def cost_function(self, X, y, weights=None, lambda_param=0.0):

想要一个可以传递给scipy.minimise的部分函数,这样我就可以找到最佳权重值(权重是一个ndarray)。

所以我需要一个只有一个参数的部分函数(x 说):

例如

cost_func(x)

但是我希望部分函数在调用原始函数时设置 weights 参数:

my_network.cost_function(X, y, weights=x, lambda_param=0.0)

(我知道我可以更改原始函数,以便权重参数是位置参数而不是关键字参数,但我不想这样做,因为我也想在没有权重参数集的情况下使用该函数)。

我认为functools.partial 满足您的需求

请参阅示例:

import functools
def cost_function(self, X, y, weights=None, lambda_param=0.0):
    print (weights)
cost_func = functools.partial(cost_function, None, 20, 30, lambda_param=2.0)

print (cost_func("Yay it worked!!!"))

最新更新