import osimport torchfrom models import Autoformer Transformer TimesNet Nonstationary_Transformer DLinear FEDformer Informer LightTS Reformer ETSformer Pyraformer PatchTST MICNclass Exp_Basicobjec
这段代码定义了一个基础实验类 Exp_Basic,包含了模型的构建、设备的获取、数据的获取、训练、验证、测试等基础方法。其中,模型的构建方法 _build_model 和数据的获取方法 _get_data 需要在子类中进行具体实现,而训练、验证、测试方法 train、vali、test 则可以根据具体需求在子类中进行重载。
该基础类还定义了一个模型字典 model_dict,其中包含了多个模型类的名称和对应的实现类,用于在子类中根据传入的模型名称获取对应的模型实例。同时,它还通过 device 属性获取当前运行设备(GPU 或 CPU)。
原文地址: https://www.cveoy.top/t/topic/bE79 著作权归作者所有。请勿转载和采集!