Closed hczs closed 8 months ago
数据集文件配置（FOGSDataset.json）：
{ "batch_size": 32, "cache_dataset": true, "num_workers": 0, "pad_with_last_sample": false, "train_rate": 0.7, "eval_rate": 0.1, "scaler": "standard", "load_external": false, "normal_external": false, "ext_scaler": "none", "input_window": 12, "output_window": 12, "add_time_in_day": false, "add_day_in_week": false, "y_start": 1, "thresh_T": 10, "direct_T": false, "weighted": false, "directed": false, "p": 1, "q": 1, "num_walks": 10, "walk_length": 25, "dimensions": 128, "window_size": 10, "workers": 0, "iter": 10, "thresh_cos": 10, "direct_L": true, "direct": false, "strides": 4 }
模型参数配置（FOGS.json）:
{ "max_epoch": 200, "learner": "adam", "learning_rate": 0.003, "lr_epsilon": 1.0e-8, "weight_decay": 0, "lr_decay": true, "lr_scheduler": "MultiStepLR", "lr_decay_ratio": 0.3, "steps": [15, 40, 70, 105, 145], "clip_grad_norm": true, "max_grad_norm": 5, "use_early_stop": true, "patience": 50, "input_window": 12, "output_window": 12, "hidden_dims": [[64, 64, 64], [64, 64, 64], [64, 64, 64]], "first_layer_embedding_size": 64, "out_layer_dim": 128, "activation": "GLU", "use_mask": true, "temporal_emb": true, "spatial_emb": true, "strides": 4, "use_trend": true, "trend_embedding": false }
跑了两个数据集，结果如下：
任务执行配置信息
数据集文件配置（FOGSDataset.json）：
模型参数配置（FOGS.json）:
任务执行结果信息
跑了两个数据集，结果如下：