如何在YAML配置文件中实现函数输出作为另一函数参数？

如果想把标签函数的配置单独拆分到另一个文件：

1. 新建label_config.yaml，内容如下：

label_generator:
  _target_: my.label.function
  _params_:
    param1: "你的第一个参数值"
    param2: "你的第二个参数值"

2. 在原数据集配置文件中导入该配置并引用函数：

# 导入外部标签配置
defaults:
  - label_config

name: DataSet
_target_: some.function.here
_params_:
  root: ${paths.datadir}
  label_name: ${call:label_generator}
  version: 0.1.0
  log_level: ${log_level}

如果你的Python脚本没有用Hydra/OmegaConf这类工具，可以手动解析YAML并执行函数：

import yaml
from my.label import function as label_func
from some.function import here as dataset_func

# 加载配置文件
with open("your_config.yaml", "r", encoding="utf-8") as f:
    config = yaml.safe_load(f)

# 先执行标签函数，获取结果
label_output = label_func(**config["label_generator"]["_params_"])

# 替换数据集配置的label_name参数
config["_params"]["label_name"] = label_output

# 初始化数据集
dataset = dataset_func(**config["_params_"])

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