Closed hczs closed 2 months ago
数据集配置文件(SimSTDataset.json)
{ "batch_size": 1024, "cache_dataset": true, "num_workers": 0, "pad_with_last_sample": true, "train_rate": 0.6, "eval_rate": 0.2, "scaler": "standard", "load_external": true, "normal_external": false, "ext_scaler": "none", "input_window": 12, "output_window": 12, "add_time_in_day": true, "add_day_in_week": true, "normalized_k": 0, "in_neighbor_num": 0, "infer_bs": 64 }
模型配置文件(SimST.json)
{ "max_epoch": 150, "learner": "adam", "learning_rate": 0.001, "weight_decay": 0.0001, "lr_decay": true, "lr_scheduler": "ReduceLROnPlateau", "lr_decay_ratio": 0.31622776601683794, "lr_patience": 10, "clip_grad_norm": true, "max_grad_norm": 5, "use_early_stop": true, "patience": 20, "node_dim": 20, "in_dim": 5, "init_dim": 64, "end_dim": 512, "layer": 2, "dropout": 0.1 }
执行器配置信息(TrafficStateExecutor.json)
{ "gpu": true, "gpu_id": 0, "max_epoch": 100, "train_loss": "none", "epoch": 0, "learner": "adam", "learning_rate": 0.01, "weight_decay": 0, "lr_epsilon": 1e-8, "lr_beta1": 0.9, "lr_beta2": 0.999, "lr_alpha": 0.99, "lr_momentum": 0, "lr_decay": false, "lr_scheduler": "multisteplr", "lr_decay_ratio": 0.1, "steps": [5, 20, 40, 70], "step_size": 10, "lr_T_max": 30, "lr_eta_min": 0, "lr_patience": 10, "lr_threshold": 1e-4, "clip_grad_norm": false, "max_grad_norm": 1.0, "use_early_stop": false, "patience": 50, "log_level": "INFO", "log_every": 1, "saved_model": true, "load_best_epoch": true, "hyper_tune": false }
模型评价指标对比如下
任务执行配置信息
数据集配置文件(SimSTDataset.json)
模型配置文件(SimST.json)
执行器配置信息(TrafficStateExecutor.json)
任务执行结果
模型评价指标对比如下