STraT代码阅读
Created|Updated
|Post Views:
Author: ALTNT
Link: http://blog.705553939.xyz/2024/11/19/sequence-processing/STraTs%E4%BB%A3%E7%A0%81%E9%98%85%E8%AF%BB/
Copyright Notice: All articles in this blog are licensed under CC BY-NC-SA 4.0 unless stating additionally.
Related Articles
Announcement
This is my Blog
Contents
- 1. STraTs 代码阅读
- 1.1. 一、数据预处理
- 1.2. 二、模型训练
- 1.2.1. 加载数据集
- 1.2.1.1. 加载刚刚保存的预训练数据：
- 1.2.1.2. 加载变量名（就是刚刚的 variable 名)
- 1.2.1.3. 从data中删除ts_id在test_ids中的行
- 1.2.1.4. 生成 val_ids
- 1.2.1.5. 生成 ts_id(即 RecorderID)到索引 id 的映射：
- 1.2.1.6. 这里就是添加了一个ts_ind列，把ts_id映射到索引，这个 ts_ind后面三元组会用
- 1.2.1.7. 保存训练集和验证集的索引
- 1.2.1.8. 正则化变量（重点)
- 1.2.1.9. 生成 不同variable 对应的唯一值
- 1.2.1.10. 初始化values, times, varis 三元组
- 1.2.1.11. 对 data 使用 minute 进行排序
- 1.2.1.12. 遍历 data 为三元组进行赋值，因为 data 已经按minute排序，所以这里可以保证 values, times, varis 中的数据是按时间顺序的
- 1.2.1.13. 去除 self.times中的重复值并保存到self.timestamps中
- 1.2.1.14. 保留入院 12 小时后的序列
- 1.2.2. 模型初始化
- 1.2.3. 设置预训练的 Evaluator
- 1.2.3.1. STraTs 数据的正向传播过程
- 1.2.1. 加载数据集