MCPcopy Create free account

hub / github.com/chaoshangcs/GTS / functions

Functions111 in github.com/chaoshangcs/GTS

↓ 13 callersMethodinverse_transform
(self, data)
lib/utils.py:64
↓ 8 callersFunctionmasked_mape_loss
(y_pred, y_true)
model/pytorch/loss.py:13
↓ 8 callersFunctionmasked_mse_loss
(y_pred, y_true)
model/pytorch/loss.py:31
↓ 7 callersFunctionmasked_mae_loss
(y_pred, y_true)
model/pytorch/loss.py:4
↓ 5 callersMethodtransform
(self, data)
lib/utils.py:61
↓ 4 callersMethod__init__
(self, temperature, logger, **model_kwargs)
model/pytorch/model.py:129
↓ 4 callersMethod_compute_loss
(self, y_true, y_predicted)
model/pytorch/supervisor.py:424
↓ 4 callersMethodevaluate
Computes mean L1Loss :return: mean L1Loss
model/pytorch/supervisor.py:138
↓ 4 callersFunctionmasked_mae_np
(preds, labels, null_val=np.nan)
lib/metrics.py:75
↓ 4 callersFunctionmasked_mape_np
(preds, labels, null_val=np.nan)
lib/metrics.py:88
↓ 4 callersFunctionmasked_rmse_np
(preds, labels, null_val=np.nan)
lib/metrics.py:58
↓ 3 callersMethod_prepare_data
(self, x, y)
model/pytorch/supervisor.py:391
↓ 3 callersMethodget_iterator
(self)
lib/utils.py:37
↓ 2 callersMethod_apply_sparse_shared
(self, grad, var, indices, scatter_add)
lib/AMSGrad.py:111
↓ 2 callersMethod_concat
(x, x_)
model/pytorch/cell.py:123
↓ 2 callersMethodget_biases
(self, length, bias_start=0.0)
model/pytorch/cell.py:22
↓ 2 callersMethodget_weights
(self, shape)
model/pytorch/cell.py:13
↓ 2 callersFunctionmasked_mse_tf
Accuracy with masking. :param preds: :param labels: :param null_val: :return:
lib/metrics.py:5
↓ 2 callersMethodtrain
(self, **kwargs)
model/pytorch/supervisor.py:134
↓ 1 callersMethod_calculate_random_walk_matrix
(self, adj_mx)
model/pytorch/cell.py:83
↓ 1 callersMethod_compute_sampling_threshold
(self, batches_seen)
model/pytorch/model.py:164
↓ 1 callersMethod_gconv
(self, inputs, adj_mx, state, output_size, bias_start=0.0)
model/pytorch/cell.py:139
↓ 1 callersMethod_get_log_dir
(kwargs)
model/pytorch/supervisor.py:78
↓ 1 callersMethod_get_x_y
:param x: shape (batch_size, seq_len, num_sensor, input_dim) :param y: shape (batch_size, horizon, num_sensor, input_dim) :re
model/pytorch/supervisor.py:396
↓ 1 callersMethod_get_x_y_in_correct_dims
:param x: shape (seq_len, batch_size, num_sensor, input_dim) :param y: shape (horizon, batch_size, num_sensor, input_dim) :re
model/pytorch/supervisor.py:411
↓ 1 callersMethod_setup_graph
(self)
model/pytorch/supervisor.py:123
↓ 1 callersMethod_train
(self, base_lr, steps, patience=200, epochs=100, lr_decay_ratio=0.1, log_every=1, save_model=0,
model/pytorch/supervisor.py:249
↓ 1 callersFunctioncalculate_normalized_laplacian
# L = D^-1/2 (D-A) D^-1/2 = I - D^-1/2 A D^-1/2 # D = diag(A 1) :param adj: :return:
lib/utils.py:85
↓ 1 callersFunctioncalculate_random_walk_matrix
(adj_mx)
lib/utils.py:101
↓ 1 callersFunctioncount_parameters
(model)
model/pytorch/model.py:8
↓ 1 callersMethoddecoder
Decoder forward pass :param encoder_hidden_state: (num_layers, batch_size, self.hidden_state_size) :param labels: (self.horiz
model/pytorch/model.py:180
↓ 1 callersMethodencoder
Encoder forward pass :param inputs: shape (seq_len, batch_size, num_sensor * input_dim) :return: encoder_hidden_state: (num_l
model/pytorch/model.py:168
↓ 1 callersFunctioneval_historical_average
(traffic_reading_df, period)
scripts/eval_baseline_methods.py:102
↓ 1 callersFunctioneval_static
(traffic_reading_df)
scripts/eval_baseline_methods.py:89
↓ 1 callersFunctioneval_var
(traffic_reading_df, n_lags=3)
scripts/eval_baseline_methods.py:114
↓ 1 callersFunctiongenerate_graph_seq2seq_io_data
Generate samples from :param df: :param x_offsets: :param y_offsets: :param add_time_in_day: :param add_day_in_week:
scripts/generate_training_data.py:12
↓ 1 callersFunctiongenerate_graph_seq2seq_io_data
Generate samples from :param df: :param x_offsets: :param y_offsets: :param add_time_in_day: :param add_day_in_week:
scripts/generate_visualization_data.py:12
↓ 1 callersFunctiongenerate_train_val_test
(args)
scripts/generate_training_data.py:56
↓ 1 callersFunctiongenerate_train_val_test
(args)
scripts/generate_visualization_data.py:61
↓ 1 callersFunctionget_adjacency_matrix
:param distance_df: data frame with three columns: [from, to, distance]. :param sensor_ids: list of sensor ids. :param normalized_k: ent
scripts/gen_adj_mx.py:11
↓ 1 callersFunctiongumbel_softmax
Sample from the Gumbel-Softmax distribution and optionally discretize. Args: logits: [batch_size, n_class] unnormalized log-probs temperatur
model/pytorch/model.py:26
↓ 1 callersFunctiongumbel_softmax_sample
(logits, temperature, eps=1e-10)
model/pytorch/model.py:21
↓ 1 callersFunctionhistorical_average_predict
Calculates the historical average of sensor reading. :param df: :param period: default 1 week. :param test_ratio: :param nul
scripts/eval_baseline_methods.py:12
↓ 1 callersMethodload_model
(self)
model/pytorch/supervisor.py:116
↓ 1 callersFunctionload_pickle
(pickle_file)
lib/utils.py:222
↓ 1 callersFunctionmain
(args)
train.py:10
↓ 1 callersFunctionmain
(args)
scripts/generate_training_data.py:106
↓ 1 callersFunctionmain
(args)
scripts/eval_baseline_methods.py:128
↓ 1 callersFunctionmain
(args)
scripts/generate_visualization_data.py:115
↓ 1 callersFunctionmasked_mae_tf
Accuracy with masking. :param preds: :param labels: :param null_val: :return:
lib/metrics.py:26
↓ 1 callersFunctionmasked_mse_np
(preds, labels, null_val=np.nan)
lib/metrics.py:62
↓ 1 callersFunctionmasked_rmse_tf
Accuracy with masking. :param preds: :param labels: :param null_val: :return:
lib/metrics.py:47
↓ 1 callersFunctionsample_gumbel
(shape, eps=1e-20)
model/pytorch/model.py:17
↓ 1 callersMethodsave_model
(self, epoch)
model/pytorch/supervisor.py:105
↓ 1 callersFunctionstatic_predict
Assumes $x^{t+1} = x^{t}$ :param df: :param n_forward: :param test_ratio: :return:
scripts/eval_baseline_methods.py:39
↓ 1 callersFunctionvar_predict
Multivariate time series forecasting using Vector Auto-Regressive Model. :param df: pandas.DataFrame, index: time, columns: sensor id, cont
scripts/eval_baseline_methods.py:53
Method__init__
:param xs: :param ys: :param batch_size: :param pad_with_last_sample: pad with the last sample to make number of sam
lib/utils.py:13
Method__init__
(self, mean, std)
lib/utils.py:57
Method__init__
(self, learning_rate=0.01, beta1=0.9, beta2=0.99, epsilon=1e-8, use_locking=False, name="AMSGrad")
lib/AMSGrad.py:16
Method__init__
(self, save_adj_name, temperature, **kwargs)
model/pytorch/supervisor.py:14
Method__init__
(self, **model_kwargs)
model/pytorch/model.py:49
Method__init__
(self, **model_kwargs)
model/pytorch/model.py:61
Method__init__
(self, **model_kwargs)
model/pytorch/model.py:95
Method__init__
(self, rnn_network: torch.nn.Module, layer_type: str)
model/pytorch/cell.py:7
Method__init__
:param num_units: :param adj_mx: :param max_diffusion_step: :param num_nodes: :param nonlinearity: :
model/pytorch/cell.py:34
Method_apply_dense
(self, grad, var)
lib/AMSGrad.py:54
Method_apply_sparse
(self, grad, var)
lib/AMSGrad.py:142
Method_build_sparse_matrix
(L)
model/pytorch/cell.py:75
Method_create_slots
(self, var_list)
lib/AMSGrad.py:31
Method_fc
(self, inputs, state, output_size, bias_start=0.0)
model/pytorch/cell.py:127
Method_finish
(self, update_ops, name_scope)
lib/AMSGrad.py:157
Method_prepare
(self)
lib/AMSGrad.py:48
Method_resource_apply_dense
(self, grad, var)
lib/AMSGrad.py:82
Method_resource_apply_sparse
(self, grad, var, indices)
lib/AMSGrad.py:153
Method_resource_scatter_add
(self, x, i, v)
lib/AMSGrad.py:148
Method_wrapper
()
lib/utils.py:40
Functionadd_simple_summary
Writes summary for a list of scalars. :param writer: :param names: :param values: :param global_step: :return:
lib/utils.py:68
Functioncalculate_metrics
Calculate the MAE, MAPE, RMSE :param df_pred: :param df_test: :param null_val: :return:
lib/metrics.py:133
Functioncalculate_reverse_random_walk_matrix
(adj_mx)
lib/utils.py:111
Functioncalculate_scaled_laplacian
(adj_mx, lambda_max=2, undirected=True)
lib/utils.py:115
Functionconfig_logging
(log_dir, log_filename='info.log', level=logging.INFO)
lib/utils.py:129
Functioncosine_similarity_torch
(x1, x2=None, eps=1e-8)
model/pytorch/model.py:11
Methodencode_onehot
(labels)
model/pytorch/model.py:149
Methodforward
Encoder forward pass. :param inputs: shape (batch_size, self.num_nodes * self.input_dim) :param hidden_state: (num_layers, ba
model/pytorch/model.py:70
Methodforward
:param inputs: shape (batch_size, self.num_nodes * self.output_dim) :param hidden_state: (num_layers, batch_size, self.hidden_state_s
model/pytorch/model.py:106
Methodforward
:param inputs: shape (seq_len, batch_size, num_sensor * input_dim) :param labels: shape (horizon, batch_size, num_sensor * output)
model/pytorch/model.py:208
Methodforward
Gated recurrent unit (GRU) with Graph Convolution. :param inputs: (B, num_nodes * input_dim) :param hx: (B, num_nodes * rnn_units)
model/pytorch/cell.py:95
Functionget_logger
(log_dir, name, log_filename='info.log', level=logging.INFO)
lib/utils.py:148
Functionget_total_trainable_parameter_size
Calculates the total number of trainable parameters in the current graph. :return:
lib/utils.py:166
Functionload_dataset
(dataset_dir, batch_size, test_batch_size=None, **kwargs)
lib/utils.py:178
Functionload_graph_data
(pkl_filename)
lib/utils.py:217
Functionloss
(preds, labels)
lib/metrics.py:103
Functionmasked_mae_loss
(scaler, null_val)
lib/metrics.py:122
Functionmasked_mse_loss
(scaler, null_val)
lib/metrics.py:102
Functionmasked_rmse_loss
(scaler, null_val)
lib/metrics.py:112
Functionmasked_rmse_loss
(y_pred, y_true)
model/pytorch/loss.py:22
Methodtest_masked_mape_np
(self)
lib/metrics_test.py:10
Methodtest_masked_mape_np2
(self)
lib/metrics_test.py:22
Methodtest_masked_mape_np_all_nan
(self)
lib/metrics_test.py:46
Methodtest_masked_mape_np_all_zero
(self)
lib/metrics_test.py:34
next →1–100 of 111, ranked by callers