MCPcopy Create free account

hub / github.com/William-Liwei/EnergyPatchTST / functions

Functions613 in github.com/William-Liwei/EnergyPatchTST

⨍ Functions 613◇ Types & classes 193
↓ 36 callersMethodabs
(self, x, y)
Formers/FEDformer/layers/utils.py:358
↓ 23 callersMethodcpu
(self)
Formers/FEDformer/layers/utils.py:317
↓ 17 callersMethodbackward
(ctx, grad_output)
Formers/Pyraformer/pyraformer/hierarchical_mm_tvm.py:264
↓ 11 callersMethodfit
(self, data)
Formers/Pyraformer/utils/tools.py:35
↓ 9 callersMethod__init__
(self, c_in, d_model, embed_type='fixed', freq='h', dropout=0.1)
Formers/FEDformer/layers/Embed.py:106
↓ 8 callersMethod__init__
(self, attn_layers, conv_layers=None, norm_layer=None)
Formers/FEDformer/layers/Autoformer_EncDec.py:123
↓ 8 callersMethod__init__
(self, opt, mask)
Formers/Pyraformer/pyraformer/Layers.py:381
↓ 8 callersMethod__init__
(self, c_in, d_model, embed_type='fixed', freq='h', dropout=0.1)
layers/Embed.py:106
↓ 7 callersMethodcuda
(self)
Formers/FEDformer/layers/utils.py:313
↓ 6 callersMethod__init__
(self, c_in, d_model, dropout=0.1)
Formers/Pyraformer/pyraformer/embed.py:82
↓ 6 callersMethod__init__
(self, attn_layers, conv_layers=None, norm_layer=None)
layers/Autoformer_EncDec.py:86
↓ 6 callersMethod__init__
(self, q_len, d_model, n_heads, d_k=None, d_v=None, d_ff=None, norm='BatchNorm', attn
layers/PatchTST_backbone.py:178
↓ 6 callersMethod_get_data
获取数据
exp/energy_exp.py:40
↓ 6 callersMethodtrain
(self, setting)
exp/exp_main.py:100
↓ 6 callersMethodtransform
(self, data)
utils/tools.py:84
↓ 5 callersMethod_get_data
(self, flag)
Formers/FEDformer/exp/exp_main.py:39
↓ 5 callersMethod_get_data
(self, flag)
exp/exp_main.py:42
↓ 5 callersMethod_prepare_tensors
Fix `stride()` information of input tensor. This addresses some inconsistency in stride information in PyTorch. For a tensor t, if t.size(0) =
Formers/Pyraformer/pyraformer/hierarchical_mm_tvm.py:205
↓ 5 callersFunctiontime_features
(dates, freq='h', timeenc=None)
utils/timefeatures.py:133
↓ 5 callersMethodtrain
(self, setting)
Formers/FEDformer/exp/exp_main.py:94
↓ 4 callersMethod__init__
(self, k=3, alpha=64, L=0, c=1, base='legendre',
Formers/FEDformer/layers/MultiWaveletCorrelation.py:297
↓ 4 callersMethod__init__
(self, attn_layers, conv_layers=None, norm_layer=None)
Formers/FEDformer/layers/Transformer_EncDec.py:54
↓ 4 callersMethod__init__
(self, attn_layers, conv_layers=None, norm_layer=None)
layers/Transformer_EncDec.py:54
↓ 4 callersMethod__init__
(self, configs)
models/Stat_models.py:103
↓ 4 callersMethod_graph_mm
Calls the compiled function after checking the input format. This function is called in three different modes. t1 x t2 = r ==> t1 and t2 are n
Formers/Pyraformer/pyraformer/hierarchical_mm_tvm.py:161
↓ 4 callersFunctionadjust_learning_rate
(optimizer, scheduler, epoch, args, printout=True)
utils/tools.py:9
↓ 4 callersMethodfit
(self, data)
Formers/Pyraformer/data_loader.py:278
↓ 4 callersMethodinverse_transform
(self, data)
Formers/Pyraformer/data_loader.py:250
↓ 4 callersMethodpredict
(self, setting, load=False)
exp/exp_main.py:315
↓ 4 callersFunctionpsi
(psi1, psi2, i, inp)
Formers/FEDformer/layers/utils.py:133
↓ 4 callersMethodtest
(self, setting, test=0)
exp/exp_main.py:219
↓ 4 callersMethodtransform
(self, data)
Formers/FEDformer/utils/tools.py:72
↓ 3 callersFunctionMSE
(pred, true)
Formers/FEDformer/utils/metrics.py:18
↓ 3 callersMethod__init__
(self, d=2, p=2, size_average=True, reduction=True)
Formers/FEDformer/layers/utils.py:347
↓ 3 callersMethod__init__
(self, mask_flag=True, factor=5, scale=None, attention_dropout=0.1, output_attention=False)
Formers/FEDformer/layers/SelfAttention_Family.py:15
↓ 3 callersMethod_build_backbone
构建PatchTST主干网络
models/EnergyPatchTST.py:95
↓ 3 callersMethod_compute_loss
计算损失
utils/pretrain_finetune.py:295
↓ 3 callersMethod_get_model_output
Get model output, handling different model architectures
utils/pretrain_finetune.py:260
↓ 3 callersMethod_get_model_output
获取模型输出
exp/energy_exp.py:68
↓ 3 callersMethod_process_model_input
处理模型输入
exp/energy_exp.py:51
↓ 3 callersFunctionget_frequency_modes
get modes on frequency domain: 'random' means sampling randomly; 'else' means sampling the lowest modes;
Formers/FEDformer/layers/FourierCorrelation.py:10
↓ 3 callersFunctionget_mask
Get the attention mask of PAM-Naive
Formers/Pyraformer/pyraformer/Layers.py:12
↓ 3 callersMethodinverse_transform
(self, data)
Formers/FEDformer/data_provider/data_loader.py:371
↓ 3 callersMethodinverse_transform
(self, data)
data_provider/data_loader.py:395
↓ 3 callersMethodreturn_result
(self)
models/Stat_models.py:29
↓ 3 callersFunctionsplit
(split_start, label, cov, pred_length)
Formers/Pyraformer/data_loader.py:494
↓ 3 callersFunctiontime_features
(dates, freq='h')
Formers/FEDformer/utils/timefeatures.py:133
↓ 3 callersFunctiontime_features
(dates, timeenc=1, freq='h')
Formers/Pyraformer/utils/timefeatures.py:114
↓ 3 callersMethodtransform
(self, data)
Formers/Pyraformer/utils/tools.py:39
↓ 3 callersMethodwavelet_transform
(self, x)
Formers/FEDformer/layers/MultiWaveletCorrelation.py:191
↓ 2 callersFunctionCoord1dPosEncoding
(q_len, exponential=False, normalize=True)
layers/PatchTST_layers.py:89
↓ 2 callersFunctionCoord2dPosEncoding
(q_len, d_model, exponential=False, normalize=True, eps=1e-3, verbose=False)
layers/PatchTST_layers.py:74
↓ 2 callersFunctionMAE
(pred, true)
Formers/FEDformer/utils/metrics.py:14
↓ 2 callersFunctionMAPE
(pred, true)
Formers/FEDformer/utils/metrics.py:26
↓ 2 callersFunctionMSE
(pred, true)
utils/metrics.py:19
↓ 2 callersFunctionMSE
(pred, true)
Formers/Pyraformer/utils/tools.py:9
↓ 2 callersFunctionMSPE
(pred, true)
Formers/FEDformer/utils/metrics.py:30
↓ 2 callersFunctionRMSE
(pred, true)
Formers/FEDformer/utils/metrics.py:22
↓ 2 callersMethod__init__
(self, opt)
Formers/Pyraformer/pyraformer/graph_attention.py:185
↓ 2 callersMethod__init__
(self, mask_flag=True, factor=5, scale=None, attention_dropout=0.1, output_attention=False)
layers/SelfAttention_Family.py:15
↓ 2 callersMethod__init__
(self, *dims, contiguous=False)
layers/PatchTST_layers.py:8
↓ 2 callersMethod__init__
(self, configs)
models/DLinear.py:42
↓ 2 callersMethod_get_lib_filename
(dtype: str, device: str)
Formers/Pyraformer/pyraformer/hierarchical_mm_tvm.py:117
↓ 2 callersMethod_select_criterion
(self)
exp/exp_main.py:50
↓ 2 callersMethod_select_optimizer
(self)
exp/exp_main.py:46
↓ 2 callersMethod_validate
验证模型
utils/pretrain_finetune.py:230
↓ 2 callersFunctioneval_epoch
Epoch operation in evaluation phase.
Formers/Pyraformer/single_step_main.py:125
↓ 2 callersFunctioneval_epoch
Epoch operation in evaluation phase.
Formers/Pyraformer/long_range_main.py:199
↓ 2 callersFunctionget_filter
(base, k)
Formers/FEDformer/layers/utils.py:131
↓ 2 callersFunctionget_k_q
Get the index of the query that can attend to the given key.
Formers/Pyraformer/pyraformer/Layers.py:153
↓ 2 callersFunctionget_q_k
Get the index of the key that a given query needs to attend to.
Formers/Pyraformer/pyraformer/Layers.py:91
↓ 2 callersFunctionmetric
(pred, true)
utils/metrics.py:35
↓ 2 callersFunctionnormalize
(inputs, seq_length)
Formers/Pyraformer/preprocess_wind.py:66
↓ 2 callersFunctionnormalize
(inputs, seq_length)
Formers/Pyraformer/preprocess_flow.py:76
↓ 2 callersFunctionprep_data
Divide the training sequence into windows
Formers/Pyraformer/preprocess_elect.py:26
↓ 2 callersFunctionprepare_dataloader
Load data and prepare dataloader.
Formers/Pyraformer/single_step_main.py:16
↓ 2 callersFunctionprepare_dataloader
Load data and prepare dataloader.
Formers/Pyraformer/long_range_main.py:14
↓ 2 callersFunctionrefer_points
Gather features from PAM's pyramid sequences
Formers/Pyraformer/pyraformer/Layers.py:50
↓ 2 callersFunctionsave
(data, v, save_dir)
Formers/Pyraformer/preprocess_wind.py:80
↓ 2 callersFunctionsave
(data, v, save_dir)
Formers/Pyraformer/preprocess_flow.py:86
↓ 2 callersMethodsave_checkpoint
(self, val_loss, model, path)
utils/tools.py:65
↓ 2 callersMethodsave_checkpoint
(self, val_loss, model, path)
Formers/FEDformer/utils/tools.py:53
↓ 2 callersMethodtest
(self, setting, test=0)
Formers/FEDformer/exp/exp_main.py:219
↓ 2 callersFunctiontest_NSA
Test the time and CUDA memory consumption of normal self attention.
Formers/Pyraformer/pyraformer/graph_attention.py:451
↓ 2 callersMethodtime_delay_agg_inference
SpeedUp version of Autocorrelation (a batch-normalization style design) This is for the inference phase.
Formers/FEDformer/layers/AutoCorrelation.py:62
↓ 2 callersMethodtime_delay_agg_training
SpeedUp version of Autocorrelation (a batch-normalization style design) This is for the training phase.
Formers/FEDformer/layers/AutoCorrelation.py:38
↓ 2 callersMethodvali
(self, vali_data, vali_loader, criterion)
Formers/FEDformer/exp/exp_main.py:51
↓ 2 callersMethodvali
验证模型
exp/energy_exp.py:240
↓ 2 callersMethodvali
(self, vali_data, vali_loader, criterion)
exp/exp_main.py:54
↓ 2 callersFunctionvisual
Results visualization
utils/tools.py:91
↓ 1 callersFunctionAE_loss
(mu, labels, ignore_zero)
Formers/Pyraformer/utils/tools.py:86
↓ 1 callersFunctionCORR
(pred, true)
utils/metrics.py:8
↓ 1 callersFunctionCORR
(pred, true)
Formers/FEDformer/utils/metrics.py:8
↓ 1 callersFunctionMAE
(pred, true)
utils/metrics.py:15
↓ 1 callersFunctionMAE
(pred, true)
Formers/Pyraformer/utils/tools.py:6
↓ 1 callersFunctionMAPE
(pred, true)
utils/metrics.py:27
↓ 1 callersFunctionMAPE
(pred, true)
Formers/Pyraformer/utils/tools.py:15
↓ 1 callersFunctionMSPE
(pred, true)
utils/metrics.py:31
↓ 1 callersFunctionMSPE
(pred, true)
Formers/Pyraformer/utils/tools.py:18
↓ 1 callersFunctionPositionalEncoding
(q_len, d_model, normalize=True)
layers/PatchTST_layers.py:61
next →1–100 of 613, ranked by callers