跟文件夹:save_models ## 按药品训练 ## 1.创建 product 文件夹 2.选择药品 product下创建药品id 文件夹,根据数据范围加上相应的后缀,聚合所有店铺all,指定店铺就店铺id 3.模型类型 对应的文件下创建模型名称的文件夹 4.在模型名称的文件夹下,版本文件夹version+第几次训练 5.在版本文件下存储对应的检查点文件,最终模型文件,损失曲线图 ## 按店铺训练 ## 1.创建 store 文件夹 2.选择店铺 store下创建店铺id 文件夹,根据药品范围加上相应的后缀,所有药品all,指定药品就药品id 3.模型类型 对应的文件下创建模型名称的文件夹 4.在模型名称的文件夹下,版本文件夹version+第几次训练 5.在版本文件下存储对应的检查点文件,最终模型文件,损失曲线图 ## 按全局训练 ## 1.创建 global 文件夹 2.选择训练范围时 创建文件夹根据数据范围,所有店铺所有药品为all,选择店铺就店铺id,选择药品就药品id ,自定义范围就根据下面的店铺id创建,再在店铺id文件夹下创建对应的药品id文件夹 3.聚合方式 根据聚合方式创建对应的文件 4.模型类型 对应的文件下创建模型名称的文件夹 5.在模型名称的文件夹下,版本文件夹version+第几次训练 6.在版本文件下存储对应的检查点文件,最终模型文件,损失曲线图 --- ## 优化后模型保存规则分析总结 与当前系统中将模型信息编码到文件名并将文件存储在相对扁平目录中的做法相比,新规则引入了一套更具结构化和层级化的模型保存策略。这种优化旨在提高模型文件的可管理性、可追溯性和可扩展性。 ### 核心思想 优化后的核心思想是**“目录即元数据”**。通过创建层级分明的目录结构,将模型的训练模式、范围、类型和版本等关键信息体现在目录路径中,而不是仅仅依赖于文件名。所有与单次训练相关的产物(最终模型、检查点、损失曲线图等)都将被统一存放在同一个版本文件夹下,便于管理和溯源。 ### 统一根目录 所有模型都将保存在 `saved_models` 文件夹下。 ### 优化后的目录结构规则 #### 1. 按药品训练 (Product Training) * **目录结构**: `saved_models/product/{product_id}_{scope}/{model_type}/v{N}/` * **路径解析**: * `product`: 表示这是按“药品”为核心的训练模式。 * `{product_id}_{scope}`: * `{product_id}`: 训练的药品ID 。 * `{scope}`: 数据的店铺范围。 * `all`: 使用所有店铺的聚合数据。 * `{store_id}`: 使用指定店铺的数据。 * `{model_type}`: 模型的类型 (例如 `mlstm`, `transformer`)。 * `v{N}`: 模型的版本号 (例如 `v1`, `v2`)。 * **文件夹内容**: * 最终模型文件 (例如 `model_final.pth`) * 训练检查点文件 (例如 `checkpoint_epoch_10.pth`, `checkpoint_best.pth`) * 损失曲线图 (例如 `loss_curve.png`) #### 2. 按店铺训练 (Store Training) * **目录结构**: `saved_models/store/{store_id}_{scope}/{model_type}/v{N}/` * **路径解析**: * `store`: 表示这是按“店铺”为核心的训练模式。 * `{store_id}_{scope}`: * `{store_id}`: 训练的店铺ID 。 * `{scope}`: 数据的药品范围。 * `all`: 使用该店铺所有药品的聚合数据。 * `{product_id}`: 使用该店铺指定药品 * `v{N}`: 模型的版本号。 * **文件夹内容**: 与“按药品训练”模式相同。 #### 3. 全局训练 (Global Training) * **目录结构**: `saved_models/global/{scope_path}/{aggregation_method}/{model_type}/v{N}/` * **路径解析**: * `global`: 表示这是“全局”训练模式。 * `{scope_path}`: 描述训练所用数据的范围,结构比较灵活: * `all`: 代表所有店铺的所有药品。 * `stores/{store_id}`: 代表选择了特定的店铺。 * `products/{product_id}`: 代表选择了特定的药品。 * `custom/{store_id}/{product_id}`: 代表自定义范围,同时指定了店铺和药品。 * `{aggregation_method}`: 数据的聚合方式 (例如 `sum`, `mean`)。 * `{model_type}`: 模型的类型。 * `v{N}`: 模型的版本号。 * **文件夹内容**: 与“按药品训练”模式相同。 ### 总结 总的来说,优化后的规则通过一个清晰、自解释的目录结构,系统化地组织了所有训练产物。这不仅使得查找和管理特定模型变得极为方便,也为未来的自动化模型管理和部署流程奠定了坚实的基础。 --- ### 优化规则下的详细文件保存、读取及数据库记录规范 基于优化后的目录结构规则,我们进一步定义详细的文件保存、读取、数据库记录及版本管理的具体规范。 #### 一、 详细文件保存路径规则 所有训练产物都保存在对应模型的版本文件夹内,并采用统一的命名约定。 * **最终模型文件**: `model.pth` * **最佳性能检查点**: `checkpoint_best.pth` * **定期检查点**: `checkpoint_epoch_{epoch_number}.pth` (例如: `checkpoint_epoch_50.pth`) * **损失曲线图**: `loss_curve.png` * **训练元数据**: `metadata.json` (包含训练参数、指标等详细信息) **示例路径:** 1. **按药品训练 (P001, 所有店铺, mlstm, v2)**: * **目录**: `saved_models/product/P001_all/mlstm/v2/` * **最终模型**: `saved_models/product/P001_all/mlstm/v2/model.pth` * **损失曲线**: `saved_models/product/P001_all/mlstm/v2/loss_curve.png` 2. **按店铺训练 (S001, 指定药品P002, transformer, v1)**: * **目录**: `saved_models/store/S001_P002/transformer/v1/` * **最终模型**: `saved_models/store/S001_P002/transformer/v1/model.pth` 3. **全局训练 (所有数据, sum聚合, kan, v5)**: * **目录**: `saved_models/global/all/sum/kan/v5/` * **最终模型**: `saved_models/global/all/sum/kan/v5/model.pth` #### 二、 文件读取规则 读取模型或其产物时,首先根据模型的元数据构建其版本目录路径,然后在该目录内定位具体文件。 **读取逻辑:** 1. **确定模型元数据**: * 训练模式 (`product`, `store`, `global`) * 范围 (`{product_id}_{scope}`, `{store_id}_{scope}`, `{scope_path}`) * 聚合方式 (仅全局模式) * 模型类型 (`mlstm`, `kan`, etc.) * 版本号 (`v{N}`) 2. **构建模型根目录路径**: 根据上述元数据拼接路径。 * *示例*: 要读取“店铺S001下P002药品的transformer模型v1”,构建路径 `saved_models/store/S001_P002/transformer/v1/`。 3. **定位具体文件**: 在构建好的目录下直接读取所需文件。 * **加载最终模型**: 读取 `model.pth`。 * **加载最佳模型**: 读取 `checkpoint_best.pth`。 * **查看损失曲线**: 读取 `loss_curve.png`。 #### 三、 数据库保存规则 数据库的核心职责是**索引模型**,而不是存储冗余信息。因此,数据库中只保存足以定位到模型版本目录的**路径**信息。 **`model_versions` 表结构优化:** | 字段名 | 类型 | 描述 | 示例 | | :--- | :--- | :--- | :--- | | `id` | INTEGER | 主键 | 1 | | `model_identifier` | TEXT | 模型的唯一标识符,由模式和范围构成 | `product_P001_all` | | `model_type` | TEXT | 模型类型 | `mlstm` | | `version` | TEXT | 版本号 | `v2` | | `model_path` | TEXT | **模型版本目录的相对路径** | `saved_models/product/P001_all/mlstm/v2/` | | `created_at` | TEXT | 创建时间 | `2025-07-15 18:40:00` | | `metrics_summary`| TEXT | 关键性能指标的JSON字符串 | `{"rmse": 10.5, "r2": 0.89}` | **保存逻辑:** * 当一次训练成功完成并生成版本 `v{N}` 后,向 `model_versions` 表中插入一条新记录。 * `model_path` 字段**只记录到版本目录**,如 `saved_models/product/P001_all/mlstm/v2/`。应用程序根据此路径和标准文件名(如 `model.pth`)来加载具体文件。 #### 四、 版本记录文件规则 为了快速、方便地获取和递增版本号,在 `saved_models` 根目录下维护一个版本记录文件。 * **文件名**: `versions.json` * **位置**: `saved_models/versions.json` * **结构**: 一个JSON对象,`key` 是模型的唯一标识符,`value` 是该模型的**最新版本号 (整数)**。 **`versions.json` 示例:** ```json { "product_P001_all_mlstm": 2, "store_S001_P002_transformer": 1, "global_all_sum_kan": 5 } ``` **版本管理流程:** 1. **获取下一个版本号**: * 在开始新训练前,根据训练参数构建模型的唯一标识符 (例如 `product_P001_all_mlstm`)。 * 读取 `saved_models/versions.json` 文件。 * 查找对应的 `key`,获取当前最新版本号。如果 `key` 不存在,则当前版本为 0。 * 下一个版本号即为 `当前版本号 + 1`。 2. **更新版本号**: * 训练成功后,将新的版本号写回到 `saved_models/versions.json` 文件中,更新对应 `key` 的 `value`。 * 这个过程需要加锁以防止并发训练时出现版本号冲突。