如何在函数中定义模型名称

Question

我想创建一个函数，在其中循环多个参数并记录我的结果。最后，我希望能够访问最佳模型。我似乎无法弄清楚的是如何在函数调用中定义模型的名称。

def save_mod(mod_name):
    mod = xgb.train(
    params,
    dtrain,
    num_boost_round=999,
    evals=[(dval, "Val")],
    early_stopping_rounds=10)
    mod_name=mod

这会运行，但是模型不会保存为我指定的名称。尝试访问示例会告诉我该示例不存在，并且如果在定义中添加引号，该示例将无法运行。

save_mod("example")

我确定这是一个基本问题，但我找不到答案。

谢谢，J

Answer 1

那是因为您没有从函数中返回任何东西。

def save_mod(mod_name):
    .
    .
    .
    return mod_name