MCPcopy Create free account

hub / github.com/kwuking/TimeMixer / functions

Functions320 in github.com/kwuking/TimeMixer

↓ 24 callersMethodload
Load cached dataset. :param training: Load training part if training is True, test part otherwise.
data_provider/m4.py:76
↓ 20 callersMethodtransform
(self, data)
utils/tools.py:75
↓ 11 callersFunctionsfb1d
1D synthesis filter bank of an image tensor
layers/DWT_Decomposition.py:472
↓ 10 callersMethod__init__
(self, c_in, d_model, embed_type='fixed', freq='h', dropout=0.1)
layers/Embed.py:110
↓ 9 callersFunctionadjust_learning_rate
(optimizer, scheduler, epoch, args, printout=True)
utils/tools.py:9
↓ 8 callersFunctionafb1d
1D analysis filter bank (along one dimension only) of an image Inputs: x (tensor): 4D input with the last two dimensions the spatial
layers/DWT_Decomposition.py:325
↓ 7 callersMethod__init__
(self, attn_layers, conv_layers=None, norm_layer=None)
layers/Autoformer_EncDec.py:114
↓ 6 callersMethod_get_data
(self, flag)
exp/exp_classification.py:36
↓ 6 callersMethodbackward
(ctx, low, highs)
layers/DWT_Decomposition.py:607
↓ 6 callersFunctionroll
(x, n, dim, make_even=False)
layers/DWT_Decomposition.py:243
↓ 5 callersMethod__init__
(self, mask_flag=True, factor=5, scale=None, attention_dropout=0.1, output_attention=False)
layers/SelfAttention_Family.py:15
↓ 5 callersMethod_get_data
(self, flag)
exp/exp_anomaly_detection.py:33
↓ 5 callersFunctiondata_provider
(args, flag)
data_provider/data_factory.py:25
↓ 5 callersMethodinverse_transform
(self, data)
utils/tools.py:78
↓ 5 callersMethodinverse_transform
(self, data)
data_provider/data_loader.py:355
↓ 5 callersMethodsummarize_groups
Re-group scores respecting M4 rules. :param scores: Scores per group. :return: Grouped scores.
utils/m4_summary.py:113
↓ 4 callersMethod__init__
(self, attn_layers, conv_layers=None, norm_layer=None)
layers/Transformer_EncDec.py:55
↓ 4 callersMethod__init__
(self, configs)
models/TimeMixer.py:188
↓ 4 callersMethod__multi_scale_process_inputs
(self, x_enc, x_mark_enc)
models/TimeMixer.py:288
↓ 4 callersMethod_get_data
(self, flag)
exp/exp_long_term_forecasting.py:29
↓ 4 callersMethod_get_data
(self, flag)
exp/exp_short_term_forecasting.py:37
↓ 4 callersMethod_get_data
(self, flag)
exp/exp_imputation.py:30
↓ 4 callersFunctionint_to_mode
(mode)
layers/DWT_Decomposition.py:549
↓ 4 callersFunctionprep_filt_afb2d
Prepares the filters to be of the right form for the afb2d function. In particular, makes the tensors the right shape. It takes mirror ima
layers/DWT_Decomposition.py:1189
↓ 4 callersFunctionreflect
Reflect the values in matrix *x* about the scalar values *minx* and *maxx*. Hence a vector *x* containing a long linearly increasing series is
layers/DWT_Decomposition.py:1242
↓ 3 callersFunctionacf
Autocorrelation function. Args: ts: time series k: lag Returns: acf value
utils/data_analysis.py:101
↓ 3 callersMethodcompl_mul1d
(self, order, x, weights)
layers/FourierCorrelation.py:104
↓ 3 callersFunctiondivide_no_nan
a/b where the resulted NaN or Inf are replaced by 0.
utils/losses.py:25
↓ 3 callersFunctionget_frequency_modes
get modes on frequency domain: 'random' means sampling randomly; 'else' means sampling the lowest modes;
layers/FourierCorrelation.py:6
↓ 3 callersFunctiongroup_values
(values, groups, group_name)
utils/m4_summary.py:28
↓ 3 callersFunctionmetric
(pred, true)
utils/metrics.py:36
↓ 3 callersFunctionmypad
Function to do numpy like padding on tensors. Only works for 2-D padding. Inputs: x (tensor): tensor to pad pad (tuple)
layers/DWT_Decomposition.py:262
↓ 3 callersMethodout_projection
(self, dec_out, i, out_res)
models/TimeMixer.py:268
↓ 3 callersFunctionprep_filt_afb1d
Prepares the filters to be of the right form for the afb2d function. In particular, makes the tensors the right shape. It takes mirror ima
layers/DWT_Decomposition.py:1220
↓ 3 callersFunctionprep_filt_sfb1d
Prepares the filters to be of the right form for the sfb1d function. In particular, makes the tensors the right shape. It does not mirror i
layers/DWT_Decomposition.py:1166
↓ 3 callersFunctiontime_features
(dates, freq='h')
utils/timefeatures.py:151
↓ 3 callersFunctionvisual
Results visualization
utils/tools.py:90
↓ 2 callersFunctionMSE
(pred, true)
utils/metrics.py:18
↓ 2 callersMethod__init__
(self)
utils/losses.py:36
↓ 2 callersMethod__init__
(self, J=1, wave='db1', mode='zero', use_amp=False)
layers/DWT_Decomposition.py:142
↓ 2 callersMethod_dummy_forward
(self, input_length)
layers/DWT_Decomposition.py:72
↓ 2 callersFunctionafb1d_atrous
1D analysis filter bank (along one dimension only) of an image without downsampling. Does the a trous algorithm. Inputs: x (tens
layers/DWT_Decomposition.py:421
↓ 2 callersFunctioncal_accuracy
(y_pred, y_true)
utils/tools.py:138
↓ 2 callersFunctionforecastabilty
Forecastability Measure. Args: ts: time series Returns: 1 - the entropy of the fourier transformation of time series / entropy
utils/data_analysis.py:6
↓ 2 callersMethodlast_insample_window
The last window of insample size of all timeseries. This function does not support batching and does not reshuffle timeseries.
data_provider/data_loader.py:358
↓ 2 callersFunctionmase
(forecast, insample, outsample, frequency)
utils/m4_summary.py:32
↓ 2 callersFunctionmode_to_int
(mode)
layers/DWT_Decomposition.py:530
↓ 2 callersFunctionprep_filt_afb2d_nonsep
Prepares the filters to be of the right form for the afb2d_nonsep function. In particular, makes 2d point spread functions, and mirror imag
layers/DWT_Decomposition.py:1065
↓ 2 callersFunctionprep_filt_sfb2d
Prepares the filters to be of the right form for the sfb2d function. In particular, makes the tensors the right shape. It does not mirror
layers/DWT_Decomposition.py:1134
↓ 2 callersFunctionprep_filt_sfb2d_nonsep
Prepares the filters to be of the right form for the sfb2d_nonsep function. In particular, makes 2d point spread functions. Does not mirror
layers/DWT_Decomposition.py:1100
↓ 2 callersMethodsave_checkpoint
(self, val_loss, model, path)
utils/tools.py:56
↓ 2 callersFunctionsmape_2
(forecast, target)
utils/m4_summary.py:36
↓ 2 callersMethodtest
(self)
exp/exp_basic.py:47
↓ 2 callersMethodvali
(self, vali_data, vali_loader, criterion)
exp/exp_classification.py:49
↓ 2 callersMethodvali
(self, vali_data, vali_loader, criterion)
exp/exp_long_term_forecasting.py:44
↓ 2 callersMethodvali
(self, vali_data, vali_loader, criterion)
exp/exp_anomaly_detection.py:45
↓ 2 callersMethodvali
(self, vali_data, vali_loader, criterion)
exp/exp_imputation.py:42
↓ 1 callersFunctionMAE
(pred, true)
utils/metrics.py:14
↓ 1 callersFunctionMAPE
(pred, true)
utils/metrics.py:26
↓ 1 callersFunctionMSPE
(pred, true)
utils/metrics.py:32
↓ 1 callersFunctionRMSE
(pred, true)
utils/metrics.py:22
↓ 1 callersMethod__init__
(self, in_channels, out_channels, stride=1, num_kernels=6, init_weight=True)
layers/Conv_Blocks.py:6
↓ 1 callersMethod__init__
(self, in_channels, out_channels, n_heads, seq_len, modes=0, mode_select_method='random')
layers/FourierCorrelation.py:25
↓ 1 callersMethod__init__
(self, mask_flag=True, factor=1, scale=None, attention_dropout=0.1, output_attention=False)
layers/AutoCorrelation.py:19
↓ 1 callersMethod__read_data__
(self)
data_provider/data_loader.py:48
↓ 1 callersMethod__read_data__
(self)
data_provider/data_loader.py:136
↓ 1 callersMethod__read_data__
(self)
data_provider/data_loader.py:226
↓ 1 callersMethod__read_data__
(self)
data_provider/data_loader.py:320
↓ 1 callersMethod__read_data__
(self)
data_provider/data_loader.py:743
↓ 1 callersMethod__read_data__
(self)
data_provider/data_loader.py:823
↓ 1 callersMethod_acquire_device
(self)
exp/exp_basic.py:19
↓ 1 callersMethod_build_model
(self)
exp/exp_basic.py:15
↓ 1 callersMethod_denormalize
(self, x)
layers/StandardNorm.py:56
↓ 1 callersMethod_get_initial_context
(self, V, L_Q)
layers/SelfAttention_Family.py:115
↓ 1 callersMethod_get_statistics
(self, x)
layers/StandardNorm.py:35
↓ 1 callersMethod_init_params
(self)
layers/DWT_Decomposition.py:81
↓ 1 callersMethod_init_params
(self)
layers/StandardNorm.py:30
↓ 1 callersMethod_initialize_weights
(self)
layers/Conv_Blocks.py:19
↓ 1 callersMethod_initialize_weights
(self)
layers/Conv_Blocks.py:50
↓ 1 callersMethod_normalize
(self, x)
layers/StandardNorm.py:43
↓ 1 callersMethod_prob_QK
(self, Q, K, sample_k, n_top)
layers/SelfAttention_Family.py:89
↓ 1 callersMethod_select_criterion
(self)
exp/exp_classification.py:45
↓ 1 callersMethod_select_criterion
(self)
exp/exp_long_term_forecasting.py:37
↓ 1 callersMethod_select_criterion
(self, loss_name='MSE')
exp/exp_short_term_forecasting.py:45
↓ 1 callersMethod_select_criterion
(self)
exp/exp_anomaly_detection.py:41
↓ 1 callersMethod_select_criterion
(self)
exp/exp_imputation.py:38
↓ 1 callersMethod_select_optimizer
(self)
exp/exp_classification.py:40
↓ 1 callersMethod_select_optimizer
(self)
exp/exp_long_term_forecasting.py:33
↓ 1 callersMethod_select_optimizer
(self)
exp/exp_short_term_forecasting.py:41
↓ 1 callersMethod_select_optimizer
(self)
exp/exp_anomaly_detection.py:37
↓ 1 callersMethod_select_optimizer
(self)
exp/exp_imputation.py:34
↓ 1 callersMethod_update_context
(self, context_in, V, scores, index, L_Q, attn_mask)
layers/SelfAttention_Family.py:128
↓ 1 callersMethod_wavelet_decompose
(self, x)
layers/DWT_Decomposition.py:85
↓ 1 callersMethod_wavelet_reverse_decompose
(self, yl, yh)
layers/DWT_Decomposition.py:102
↓ 1 callersFunctionadjustment
(gt, pred)
utils/tools.py:114
↓ 1 callersMethodanomaly_detection
(self, x_enc)
models/TimeMixer.py:435
↓ 1 callersMethodclassification
(self, x_enc, x_mark_enc)
models/TimeMixer.py:409
↓ 1 callersFunctioncollate_fn
Build mini-batch tensors from a list of (X, mask) tuples. Mask input. Create Args: data: len(batch_size) list of tuples (X, y).
data_provider/uea.py:7
↓ 1 callersMethodcompl_mul1d
(self, order, x, weights)
layers/FourierCorrelation.py:46
↓ 1 callersMethodevaluate
Evaluate forecasts using M4 test dataset. :param forecast: Forecasts. Shape: timeseries, time. :return: sMAPE and OWA groupe
utils/m4_summary.py:57
next →1–100 of 320, ranked by callers