Vega 开源项目使用教程

1. 项目的目录结构及介绍

Vega 是一个由华为诺亚方舟实验室开发的 AutoML 工具链。其目录结构如下：

vega/
├── docs/
│   ├── 文档相关文件
├── evaluate_service/
│   ├── 评估服务相关文件
├── examples/
│   ├── 示例代码
├── vega/
│   ├── 核心代码
├── LICENSE
├── MANIFEST.in
├── README.cn.md
├── README.md
├── RELEASE.md
├── Third_Party_Open_Source_Notice
├── setup.py

目录结构介绍

• docs/: 包含项目的文档文件，如安装指南、配置指南等。
• evaluate_service/: 包含评估服务的相关代码。
• examples/: 包含多个示例代码，展示了如何使用 Vega 进行 AutoML 任务。
• vega/: 项目的核心代码，包含了 AutoML 的主要功能实现。
• LICENSE: 项目的开源许可证文件。
• MANIFEST.in: 用于指定在打包时需要包含的文件。
• README.cn.md: 项目的中文介绍文档。
• README.md: 项目的英文介绍文档。
• RELEASE.md: 项目的发布说明。
• Third_Party_Open_Source_Notice: 第三方开源软件的声明文件。
• setup.py: 用于安装项目的 Python 脚本。

2. 项目的启动文件介绍

Vega 项目的启动文件主要是 setup.py 和 vega 命令。

setup.py

setup.py 是一个标准的 Python 安装脚本，用于安装 Vega 项目。你可以通过以下命令来安装 Vega：

pip3 install --user --upgrade noah-vega

vega 命令

安装完成后，你可以使用 vega 命令来启动 Vega 应用程序。例如，运行 CARS 算法：

vega /examples/nas/cars/cars.yml

cars.yml 文件包含了管道、搜索算法、搜索空间和训练参数的定义。

3. 项目的配置文件介绍

Vega 的配置文件主要是一个 YAML 文件，用于定义 AutoML 任务的各个参数。以下是一个示例配置文件的内容：

pipeline:
  - name: "search"
    type: "SearchAlgorithm"
    parameters:
      algorithm: "CARS"
      search_space: "examples/nas/cars/cars.py"
      trainer: "examples/nas/cars/trainer.py"
      evaluator: "examples/nas/cars/evaluator.py"

配置文件介绍

• pipeline: 定义了 AutoML 任务的管道。
  • name: 管道的名称。
  • type: 管道的类型，如 SearchAlgorithm。
  • parameters: 管道的参数。
    • algorithm: 使用的搜索算法，如 CARS。
    • search_space: 搜索空间的定义文件。
    • trainer: 训练器的定义文件。
    • evaluator: 评估器的定义文件。

通过配置文件，你可以灵活地定义 AutoML 任务的各个环节，从而实现自动化的机器学习流程。