如何适配接收NamedTuple参数的目标函数与Scikit Optimize的@use_named_args

原代码报错的核心原因 #

@use_named_args的运行逻辑是：读取搜索空间每个维度的name属性，把搜索得到的参数值按“参数名=参数值”的形式，作为独立关键字传给被装饰的函数。
原代码存在两个关键问题：

1. 被装饰的my_objective_function签名根本没有foo/bar/baz这几个入参，只接收v_args、f_args两个对象，装饰器传参时自然会报“接收到未定义的关键字参数”的错误。
2. partial的套用顺序不对，先给函数加装饰器再套partial，会打乱装饰器本身的参数解析逻辑。

可正常运行的适配实现 #

整个逻辑不需要修改项目要求的“核心目标函数必须接收NamedTuple入参”的接口规范，只需要加一层薄的桥接逻辑做参数转换：

1. 把原来带NamedTuple入参的计算逻辑拆成独立的核心函数，完全保留原有签名，不做任何改动。
2. 写一个简单的桥接函数，给这个桥接函数加@use_named_args装饰，让它接收skopt传来的独立超参数值。
3. 在桥接函数内部，把收到的独立超参数组装成项目要求的variable_params NamedTuple，再调用核心计算函数。
4. 用partial给桥接函数绑定固定的参数值，作为最终传给搜索器的目标函数。

from typing import NamedTuple
from skopt.space import Real
from skopt import forest_minimize
from skopt.utils import use_named_args
from functools import partial

# 定义搜索维度
dim1 = Real(name='foo', low=0.0, high=1.0)
dim2 = Real(name='bar', low=0.0, high=1.0)
dim3 = Real(name='baz', low=0.0, high=1.0)
dimensions = [dim1, dim2, dim3]

# 定义项目要求的NamedTuple参数结构
class variable_params(NamedTuple):
    bar: float
    foo: float
    baz: float

class fixed_params(NamedTuple):
    bar1: int
    foo1: int
    baz1: int

# 实例化固定参数对象
fixed_args = fixed_params(foo1=2, bar1=3, baz1=4)

# 核心业务计算函数，完全遵循项目接口规范，入参为两个NamedTuple，不做任何修改
def core_objective(v_args: variable_params, f_args: fixed_params):
    return v_args.foo ** 2 + v_args.bar ** 4 + v_args.baz ** 8 + f_args.foo1 * 2 + f_args.bar1 * 4 + f_args.baz1 * 8

# 桥接层：用@use_named_args装饰该函数，接收搜索器传来的独立参数
@use_named_args(dimensions=dimensions)
def objective_bridge(foo: float, bar: float, baz: float, f_args: fixed_params):
    # 把独立参数组装成要求的variable_params NamedTuple
    v_args = variable_params(foo=foo, bar=bar, baz=baz=baz)
    # 调用核心业务函数
    return core_objective(v_args, f_args)

# 给桥接函数绑定固定参数，作为搜索器的入参
search_func = partial(objective_bridge, f_args=fixed_args)

# 执行超参搜索
result = forest_minimize(
    func=search_func,
    dimensions=dimensions,
    n_calls=20,
    base_estimator="ET",
    random_state=4
)

常见踩坑提醒 #

• 千万不要把@use_named_args直接加在要遵守接口规范的核心函数上，否则会把函数签名强制改成接收独立超参数，直接违反项目接口要求。桥接层的作用就是把skopt的传参规则和项目的接口规范隔离开，两边都不需要为对方修改。
• 固定参数的partial绑定必须在桥接函数装饰完成之后再做，如果先套partial再加装饰器，装饰器会把partial封装的参数当成要解析的搜索参数，直接运行报错。
• 搜索维度定义里的参数名，必须和桥接函数的入参名完全一致，不然装饰器做关键字传参时会匹配失败。

内容的提问来源于stack exchange，提问作者eng2019