当前位置：首页 > news >正文

时间序列预测（十八）——实现配置管理和扩展命令行参数解析器

news 2025/6/19 16:05:33

如图，这是一个main,py文件，在此代码中，最开始定义了许多模型参数，为了使项目更加灵活和可扩展，便于根据不同的需求调整参数和配置，可以根据实际需要扩展参数和配置项。

下面是如何实现配置管理和扩展命令行参数解析器的具体建议：

一、配置管理（使用JSON或YAML文件）

可以将配置参数存储在JSON或YAML文件中，然后在脚本中读取这些配置。以下是两个示例。

1、使用JSON文件

首先，创建一个config.json文件，内容如下：

{"feature_columns": [2, 3, 4, 5, 6, 7, 8],"label_columns": [4, 5],"predict_day": 1,"input_size": 7,"output_size": 2,"hidden_size": 128,"lstm_layers": 2,"dropout_rate": 0.2,"time_step": 20,"do_train": true,"do_predict": true,"train_data_path": "./data/stock_data.csv","model_save_path": "./checkpoint/pytorch/","log_save_path": "./log/"
}

然后，在你的脚本中使用以下代码加载JSON配置：

import jsonclass Config:def __init__(self, config_file):with open(config_file, 'r') as f:config_data = json.load(f)for key, value in config_data.items():setattr(self, key, value)# 使用示例
# config = Config('config.json')
# print(config.feature_columns)

2、使用YAML文件

首先，安装PyYAML库（如果尚未安装）：

pip install pyyaml

然后，创建一个config.yaml文件，内容如下：

feature_columns: [2, 3, 4, 5, 6, 7, 8]
label_columns: [4, 5]
predict_day: 1
input_size: 7
output_size: 2
hidden_size: 128
lstm_layers: 2
dropout_rate: 0.2
time_step: 20
do_train: true
do_predict: true
train_data_path: ./data/stock_data.csv
model_save_path: ./checkpoint/pytorch/
log_save_path: ./log/

然后，在你的脚本中使用以下代码加载YAML配置：

import yamlclass Config:def __init__(self, config_file):with open(config_file, 'r') as f:config_data = yaml.safe_load(f)for key, value in config_data.items():setattr(self, key, value)# 使用示例
# config = Config('config.yaml')
# print(config.feature_columns)

二、扩展命令行参数解析器

使用 argparse 模块扩展命令行参数解析：

import argparse  # 导入 argparse 模块，用于解析命令行参数
from config import Config  # 从 config 模块导入 Config 类，用于加载配置文件def parse_args():# 创建一个 ArgumentParser 对象，用于处理命令行参数parser = argparse.ArgumentParser(description="Your Project Description")# 添加 --config 参数，接受配置文件路径，默认为 'config.json'parser.add_argument('--config', type=str, default='config.json', help='Path to config file (JSON or YAML)')# 添加 --train 参数，作为布尔值，指示是否训练模型parser.add_argument('--train', action='store_true', help='Train the model')# 添加 --predict 参数，作为布尔值，指示是否进行预测parser.add_argument('--predict', action='store_true', help='Make predictions')# 解析命令行参数并返回return parser.parse_args()def main():args = parse_args()  # 调用 parse_args() 函数解析命令行参数config = Config(args.config)  # 根据命令行提供的配置文件路径加载配置# 如果命令行参数中包含 --train 或配置中 do_train 为 True，则开始训练if args.train or config.do_train:print("Training with parameters:")  # 输出正在训练的提示print(f"Feature columns: {config.feature_columns}")  # 打印特征列print(f"Learning rate: {config.hidden_size}")  # 打印隐藏层大小（作为学习率的示例）# 如果命令行参数中包含 --predict 或配置中 do_predict 为 True，则进行预测if args.predict or config.do_predict:print("Making predictions...")  # 输出正在进行预测的提示if __name__ == "__main__":main()  # 当脚本被直接运行时，调用 main() 函数