随笔分类 -  Informer源码学习记录

针对Informer模型，对其进行源代码解读，梳理其中的参数设置，数据集处理、模型搭建以及模型训练、测试和预测的处理流程。
Informer源代码学习记录之“数据集处理handle”
摘要：一、源码代码核心 exp_informer.py: # 获取数据集处理类, 选取的数据集 为'weather'，数据集处理类Dataset_Custom Data = data_dict[self.args.data] # DataLoader数据加载器,对数据进行 batch 的划分。结合了数据集 阅读全文
posted @ 2023-01-24 21:09 才品 阅读(690) 评论(0) 推荐(0)
Informer源码学习记录之 "模型搭建build_model"（模型初始化）
摘要：一、初始化 1. 代码结构 main_informer.py: exp = Exp(args) # set experiments exp_basic.py: class Exp_Informer(Exp_Basic): class Exp_Basic(object): def __init__(s 阅读全文
posted @ 2023-01-24 17:46 才品 阅读(1075) 评论(0) 推荐(0)
Informer源代码学习记录之“参数配置param”
摘要：一、参数配置 # 创建解析器ArgumentParser() 对象， description = [informer] 长序列预测 parser = argparse.ArgumentParser(description='[Informer] Long Sequences Forecasting' 阅读全文
posted @ 2023-01-13 20:08 才品 阅读(4008) 评论(0) 推荐(0)