### 工作原理 该代码的主要功能是从指定的 YAML 文件中加载配置数据并打印其内容。以下是具体工作流程： 1. **导入模块** 使用 `ruamel` 模块中的 `yaml` 来解析 YAML 文件的内容[^1]。 2. **打开文件** 打开指定路径下的 YAML 文件，编码方式设置为 UTF-8，确保支持多语言字符集[^4]。 3. **读取与关闭文件** 将文件内容读入内存后立即关闭文件流，防止资源泄漏。 4. **加载 YAML 数据** 调用 `yaml.load()` 方法将 YAML 文本转换成 Python 字典结构。 5. **打印数据** 输出加载后的字典数据以便调试或进一步处理。 6. **路径构建** 利用当前脚本所在的绝对路径动态拼接目标 YAML 配置文件的位置[^3]。 --- ### 用户需自定义的部分 #### 1. **YAML 文件路径** - 变量 `current_path` 是通过 `os.path.abspath(".")` 获取到的当前目录。 - 如果 YAML 文件不在程序运行目录下，则需要调整变量 `yaml_path` 中的目标路径。 #### 2. **data_dict['train'] 和 data_dict['val'] 的作用** - 假设加载的 YAML 文件是一个训练配置文件，通常会包含如下字段： ```yaml train: path/to/train/dataset val: path/to/validation/dataset ``` - 这里的 `data_dict['train']` 表示训练数据集的路径，而 `data_dict['val']` 表示验证数据集的路径。 - 用户可以根据实际需求修改这些键对应的值，指向具体的本地数据存储位置。 #### 3. **其他可能的自定义项** - 若 YAML 文件还包含了额外的关键字（如模型参数、超参等），也需要根据实际情况进行填充或校正[^2]。 --- ### 示例代码改进版 为了增强代码的安全性和易维护性，可以稍加改造原函数： ```python import os from ruamel import yaml def load_yaml_data(yaml_file): try: with open(yaml_file, 'r', encoding='utf-8') as file: data = yaml.load(file.read(), Loader=yaml.Loader) return data except FileNotFoundError: print(f"Error: File {yaml_file} not found.") return None except Exception as e: print(f"An error occurred while loading the YAML file: {e}") return None if __name__ == "__main__": current_path = os.path.abspath(".") yaml_path = os.path.join(current_path, "dict_config.yaml") config_data = load_yaml_data(yaml_path) if config_data is not None and isinstance(config_data, dict): print("Training dataset:", config_data.get('train')) print("Validation dataset:", config_data.get('val')) ``` 上述版本增加了异常捕获机制，并提供了更清晰的日志输出。 --- ###