MCPcopy Create free account

hub / github.com/EternityZY/STNSCM / functions

Functions64 in github.com/EternityZY/STNSCM

⨍ Functions 64◇ Types & classes 18
↓ 4 callersMethod__init__
(self, c_in, c_out, gdep, dropout_prob, graph_num, type=None)
model/GCN.py:70
↓ 4 callersFunctionmasked_mae_torch
(preds, labels, null_val=np.nan)
tools/metrics.py:44
↓ 4 callersFunctionmasked_mape_torch
(preds, labels, null_val=np.nan)
tools/metrics.py:58
↓ 4 callersFunctionmasked_rmse_torch
(preds, labels, null_val=np.nan)
tools/metrics.py:40
↓ 4 callersFunctionrad
(d)
tools/utils.py:149
↓ 3 callersMethodeval
(self, input, input_time, target, target_time)
model/helper.py:191
↓ 3 callersMethodinverse_transform
(self, data)
tools/utils.py:80
↓ 3 callersMethodtransform
(self, data)
tools/utils.py:77
↓ 2 callersMethod__generate_one_hot
(self, arr)
model/datasets.py:171
↓ 2 callersFunctionmetric
(pred, real)
tools/metrics.py:78
↓ 2 callersFunctionmodel_test
(runid, engine, dataloader, device, logger, cfg, mode='Test')
model/tester.py:81
↓ 2 callersFunctionmodel_val
(runid, engine, dataloader, device, logger, epoch)
model/tester.py:38
↓ 1 callersFunctionDTW
(M1, M2)
tools/utils.py:165
↓ 1 callersFunction__distance
(w1, w2)
tools/utils.py:172
↓ 1 callersMethod_compute_sampling_threshold
(self, global_step)
model/Decoder.py:135
↓ 1 callersFunctionasym_adj
Asymmetrically normalize adjacency matrix.
tools/utils.py:136
↓ 1 callersFunctionbaseline_test
(runid, model, dataloader, device, logger, cfg)
model/tester.py:165
↓ 1 callersFunctionbaseline_train
(runid, model, model_name, dataloader, static_norm_ad
model/trainer.py:40
↓ 1 callersFunctiongetDistance
(point1, point2)
tools/utils.py:146
↓ 1 callersFunctionget_logger
设置Log格式 :param log_dir: :param name: :param log_filename: :param level: :param write_to_file: :return:
config/config.py:48
↓ 1 callersMethodget_lr
(self)
model/helper.py:59
↓ 1 callersMethodinitHidden
建立RNN初始出入的隐藏层，该隐藏层不需要计算梯度 :param batch_size: :param num_nodes: :param hidden_dim: :return:
model/Encoder.py:138
↓ 1 callersMethodinput_process
(self, x, x_time, fusion_mode)
model/GraphGateRNN.py:111
↓ 1 callersFunctionload_dataset
(dataset_dir, train_batch_size, valid_batch_size=None, test
model/datasets.py:33
↓ 1 callersFunctionmasked_mse_torch
(preds, labels, null_val=np.nan)
tools/metrics.py:26
↓ 1 callersMethodtrain
(self, input, input_time, target, target_time, target_cl)
model/helper.py:118
↓ 1 callersMethodzoneout
model/GraphGateRNN.py:234
Method__getitem__
(self, item)
model/datasets.py:157
Method__init__
:optimizer: TODO :milestones: TODO :gamma: TODO :last_epoch: TODO :min_lr: TODO
tools/utils.py:34
Method__init__
(self, mean, std)
tools/utils.py:73
Method__init__
(self, max, min)
tools/utils.py:95
Method__init__
(self, mean, std, device)
tools/utils.py:115
Method__init__
(self, in_channels, time_channels, hidden_channels, gcn_depth, alpha, num_of_weeks, num_of_da
model/Encoder.py:41
Method__init__
(self, input_dim, time_dim, hidden_dim, output_dim, gcn_depth, fusion_mode, num_of_weeks, num
model/STNSCN.py:33
Method__init__
(self, in_channels, time_channels, hidden_channels, dropout
model/GraphGateRNN.py:35
Method__init__
(self, in_channels, time_channels, hidden_channels, output_channels, gcn_depth, alpha, num_of
model/Decoder.py:31
Method__init__
(self, time_channels, hidden_channels, num_of_weeks, num_of_days, num_of_hours, num_for_predi
model/Transform.py:27
Method__init__
(self, x, x_time, target, target_time, pos, target_cl=None)
model/datasets.py:141
Method__init__
(self, in_channels, hidden_channels, dropout_prob, node_num, reduction=16, alpha=3, norm='D-1')
model/DynamicGraph.py:30
Method__init__
(self)
model/GCN.py:30
Method__init__
(self)
model/GCN.py:40
Method__init__
(self)
model/GCN.py:50
Method__init__
(self, c_in, c_out, bias=True)
model/GCN.py:61
Method__init__
:optimizer: TODO :milestones: TODO :gamma: TODO :last_epoch: TODO :min_lr: TODO
model/helper.py:37
Method__init__
(self, model, base_lr, weight_decay, miles
model/helper.py:72
Method__len__
(self)
model/datasets.py:168
Methodforward
:param input: [batch, node, num_pred*3, input_dim] :param x_time: [batch, node, num_pred*3, time_dim] :param seq_length: num
model/Encoder.py:81
Methodforward
(self, x, x_time, target_time, target_cl=None, task_level=2, global_step=None)
model/STNSCN.py:106
Methodforward
(self, input, input_time, Hidden_State, encoder_hidden=None)
model/GraphGateRNN.py:176
Methodforward
:param decoder_input: [batch_size, node_num, time_len=1, in_channels] :param target_time: [batch_size, node_num, num_for_target, t
model/Decoder.py:75
Methodforward
:param encoder_hidden: [batch_size, rnn_layer, node_num, num_for_predict, hidden_channels] :param x_time: [batch, node_num, 18, 67]
model/Transform.py:49
Methodforward
(self, input, hidden)
model/DynamicGraph.py:52
Methodforward
(self, A, x)
model/GCN.py:33
Methodforward
(self, A, x)
model/GCN.py:43
Methodforward
(self, A, x)
model/GCN.py:53
Methodforward
(self, x)
model/GCN.py:65
Methodforward
(self, x, norm_adj, dyn_norm_adj=None)
model/GCN.py:96
Methodget_lr
(self)
tools/utils.py:56
Methodinverse_transform
(self, data)
tools/utils.py:104
Methodinverse_transform
(self, data)
tools/utils.py:122
Functionread_cfg_file
读取配置文件 :param filename: :return:
config/config.py:37
Functionsym_adj
Symmetrically normalize adjacency matrix.
tools/utils.py:126
Methodtransform
(self, data)
tools/utils.py:99
Methodtransform
(self, data)
tools/utils.py:119