Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/DAMO-DI-ML/NeurIPS2023-One-Fits-All
/ functions
Functions
1,009 in github.com/DAMO-DI-ML/NeurIPS2023-One-Fits-All
⨍
Functions
1,009
◇
Types & classes
319
↓ 1 callers
Function
adjust_learning_rate
(optimizer, epoch, args)
Short-term_Forecasting/utils/tools.py:8
↓ 1 callers
Function
adjust_learning_rate
(optimizer, epoch, args)
Anomaly_Detection/utils/tools.py:8
↓ 1 callers
Function
adjust_learning_rate
(optimizer, epoch, args)
Imputation/utils/tools.py:8
↓ 1 callers
Function
adjustment
(gt, pred)
Anomaly_Detection/utils/tools.py:91
↓ 1 callers
Method
analyze_classification
For an array of label predictions and the respective true labels, shows confusion matrix, accuracy, recall, precision etc: Input:
Classification/src/utils/analysis.py:409
↓ 1 callers
Method
anomaly_detection
(self, x_enc)
Short-term_Forecasting/models/GPT4TS.py:160
↓ 1 callers
Method
anomaly_detection
(self, x_enc)
Anomaly_Detection/models/GPT4TS.py:160
↓ 1 callers
Method
anomaly_detection
(self, x_enc)
Imputation/models/GPT4TS.py:160
↓ 1 callers
Method
build_output_module
(self, d_model, max_len, num_classes)
Classification/src/models/ts_transformer.py:282
↓ 1 callers
Function
check_progress
(epoch)
Classification/src/running.py:228
↓ 1 callers
Method
classification
(self, x_enc, x_mark_enc)
Short-term_Forecasting/models/GPT4TS.py:208
↓ 1 callers
Method
classification
(self, x_enc, x_mark_enc)
Anomaly_Detection/models/GPT4TS.py:208
↓ 1 callers
Method
classification
(self, x_enc, x_mark_enc)
Imputation/models/GPT4TS.py:208
↓ 1 callers
Function
collate_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).
Short-term_Forecasting/data_provider/uea.py:7
↓ 1 callers
Function
collate_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).
Anomaly_Detection/data_provider/uea.py:7
↓ 1 callers
Function
collate_unsuperv
Build mini-batch tensors from a list of (X, mask) tuples. Mask input. Create Args: data: len(batch_size) list of tuples (X, mask).
Classification/src/datasets/dataset.py:193
↓ 1 callers
Function
compensate_masking
Compensate feature vectors after masking values, in a way that the matrix product W @ X would not be affected on average. If p is the proport
Classification/src/datasets/dataset.py:175
↓ 1 callers
Method
compl_mul1d
(self, order, x, weights)
Zero-shot_Learning/layers/MultiWaveletCorrelation.py:474
↓ 1 callers
Method
compl_mul1d
(self, input, weights)
Zero-shot_Learning/layers/FourierCorrelation.py:45
↓ 1 callers
Function
convert_metrics_per_batch_to_per_sample
Args: metrics: list of len(num_batches), each element: list of len(num_metrics), each element: (num_active_in_batch,) metric per element
Classification/src/running.py:145
↓ 1 callers
Function
convert_tsf_to_dataframe
( full_file_path_and_name, replace_missing_vals_with="NaN", value_column_name="series_value", )
Zero-shot_Learning/utils/tools.py:107
↓ 1 callers
Function
convert_tsf_to_dataframe
( full_file_path_and_name, replace_missing_vals_with="NaN", value_column_name="series_value", )
Few-shot_Learning/utils/tools.py:106
↓ 1 callers
Function
convert_tsf_to_dataframe
( full_file_path_and_name, replace_missing_vals_with="NaN", value_column_name="series_value", )
Long-term_Forecasting/utils/tools.py:106
↓ 1 callers
Function
data_provider
(args, flag)
Short-term_Forecasting/data_provider/data_factory.py:22
↓ 1 callers
Function
data_provider
(args, flag)
Anomaly_Detection/data_provider/data_factory.py:22
↓ 1 callers
Function
data_provider
(args, flag)
Imputation/data_provider/data_factory.py:22
↓ 1 callers
Function
evaluate
Perform a single, one-off evaluation on an evaluator object (initialized with a dataset)
Classification/src/running.py:167
↓ 1 callers
Method
evaluate
Evaluate forecasts using M4 test dataset. :param forecast: Forecasts. Shape: timeseries, time. :return: sMAPE and OWA groupe
Short-term_Forecasting/utils/m4_summary.py:57
↓ 1 callers
Method
evenOdd
(self, x)
Zero-shot_Learning/layers/MultiWaveletCorrelation.py:381
↓ 1 callers
Method
evenOdd
(self, x)
Zero-shot_Learning/layers/MultiWaveletCorrelation.py:576
↓ 1 callers
Method
extrapolate
(self, x_freq, f, t)
Zero-shot_Learning/layers/ETSformer_EncDec.py:160
↓ 1 callers
Method
factory
(split_type, *args, **kwargs)
Classification/src/datasets/datasplit.py:57
↓ 1 callers
Method
fit_length
(self, queries)
Zero-shot_Learning/layers/SelfAttention_Family.py:229
↓ 1 callers
Method
forecast
(self, x_enc, x_mark_enc, x_dec, x_mark_dec)
Short-term_Forecasting/models/GPT4TS.py:119
↓ 1 callers
Method
forecast
(self, x_enc, x_mark_enc, x_dec, x_mark_dec)
Anomaly_Detection/models/GPT4TS.py:119
↓ 1 callers
Method
forecast
(self, x_enc, x_mark_enc, x_dec, x_mark_dec)
Imputation/models/GPT4TS.py:119
↓ 1 callers
Method
generate_classification_report
Returns a string of a report for given metric arrays (array length equals the number of classes). Called internally by `analyze_class
Classification/src/utils/analysis.py:274
↓ 1 callers
Method
get_avg_prec_recall
Get average recall and precision, using class frequencies as weights, optionally excluding specified classes
Classification/src/utils/analysis.py:340
↓ 1 callers
Method
get_exponential_weight
(self, T)
Zero-shot_Learning/layers/ETSformer_EncDec.py:70
↓ 1 callers
Function
get_loss_module
(config)
Classification/src/models/loss.py:6
↓ 1 callers
Function
get_optimizer
(name)
Classification/src/optimizers.py:6
↓ 1 callers
Function
get_phi_psi
(k, base)
Zero-shot_Learning/layers/MultiWaveletCorrelation.py:31
↓ 1 callers
Method
imputation
(self, x_enc, x_mark_enc, x_dec, x_mark_dec, mask)
Short-term_Forecasting/models/GPT4TS.py:91
↓ 1 callers
Method
imputation
(self, x_enc, x_mark_enc, x_dec, x_mark_dec, mask)
Anomaly_Detection/models/GPT4TS.py:91
↓ 1 callers
Method
imputation
(self, x_enc, x_mark_enc, x_dec, x_mark_dec, mask)
Imputation/models/GPT4TS.py:91
↓ 1 callers
Method
instance_norm
(self, case)
Short-term_Forecasting/data_provider/data_loader.py:1126
↓ 1 callers
Method
instance_norm
(self, case)
Anomaly_Detection/data_provider/data_loader.py:1126
↓ 1 callers
Method
instance_norm
(self, case)
Imputation/data_provider/data_loader.py:1126
↓ 1 callers
Method
jitter
(self, x)
Zero-shot_Learning/layers/ETSformer_EncDec.py:18
↓ 1 callers
Function
legendreDer
(k, x)
Zero-shot_Learning/layers/MultiWaveletCorrelation.py:16
↓ 1 callers
Function
limiter
Wraps a list of metric functions, i.e precison or recall, by ingoring predictions under the specified threshold for a specific class.
Classification/src/utils/analysis.py:80
↓ 1 callers
Method
load_all
Loads datasets from csv files contained in `root_dir` into a dataframe, optionally choosing from `pattern` Args: root_dir
Classification/src/datasets/data.py:134
↓ 1 callers
Method
load_all
Loads datasets from csv files contained in `root_dir` into a dataframe, optionally choosing from `pattern` Args: root_dir
Classification/src/datasets/data.py:246
↓ 1 callers
Method
load_all
Loads datasets from csv files contained in `root_dir` into a dataframe, optionally choosing from `pattern` Args: root_dir
Classification/src/datasets/data.py:386
↓ 1 callers
Method
load_all
Loads datasets from csv files contained in `root_path` into a dataframe, optionally choosing from `pattern` Args: root_pa
Short-term_Forecasting/data_provider/data_loader.py:1062
↓ 1 callers
Method
load_all
Loads datasets from csv files contained in `root_path` into a dataframe, optionally choosing from `pattern` Args: root_pa
Anomaly_Detection/data_provider/data_loader.py:1062
↓ 1 callers
Method
load_all
Loads datasets from csv files contained in `root_path` into a dataframe, optionally choosing from `pattern` Args: root_pa
Imputation/data_provider/data_loader.py:1062
↓ 1 callers
Method
load_single
(self, filepath)
Classification/src/datasets/data.py:281
↓ 1 callers
Method
load_single
(self, filepath)
Short-term_Forecasting/data_provider/data_loader.py:1090
↓ 1 callers
Method
load_single
(self, filepath)
Anomaly_Detection/data_provider/data_loader.py:1090
↓ 1 callers
Method
load_single
(self, filepath)
Imputation/data_provider/data_loader.py:1090
↓ 1 callers
Function
main
(config)
Classification/src/main.py:39
↓ 1 callers
Function
mape
(forecast, target)
Short-term_Forecasting/utils/m4_summary.py:43
↓ 1 callers
Function
mape
(forecast, target)
Anomaly_Detection/utils/m4_summary.py:43
↓ 1 callers
Function
mape
(forecast, target)
Imputation/utils/m4_summary.py:43
↓ 1 callers
Function
metric
(pred, true)
Few-shot_Learning/utils/metrics.py:40
↓ 1 callers
Function
metric
(pred, true)
Long-term_Forecasting/utils/metrics.py:40
↓ 1 callers
Function
metric
(pred, true)
Imputation/utils/metrics.py:34
↓ 1 callers
Function
noise_mask
Creates a random boolean mask of the same shape as X, with 0s at places where a feature should be masked. Args: X: (seq_length, feat_
Classification/src/datasets/dataset.py:234
↓ 1 callers
Method
normalize
Args: df: input dataframe Returns: df: normalized dataframe
Short-term_Forecasting/data_provider/uea.py:77
↓ 1 callers
Method
normalize
Args: df: input dataframe Returns: df: normalized dataframe
Anomaly_Detection/data_provider/uea.py:77
↓ 1 callers
Function
padding_mask
Used to mask padded positions: creates a (batch_size, max_len) boolean mask from a tensor of sequence lengths, where 1 means keep element at
Short-term_Forecasting/data_provider/uea.py:44
↓ 1 callers
Function
padding_mask
Used to mask padded positions: creates a (batch_size, max_len) boolean mask from a tensor of sequence lengths, where 1 means keep element at
Anomaly_Detection/data_provider/uea.py:44
↓ 1 callers
Function
padding_mask
Used to mask padded positions: creates a (batch_size, max_len) boolean mask from a tensor of sequence lengths, where 1 means keep element at
Imputation/data_provider/uea.py:44
↓ 1 callers
Method
parse
(self)
Classification/src/options.py:170
↓ 1 callers
Method
prec_rec_histogram
Make a histogram with the distribution of classes with respect to precision and recall
Classification/src/utils/analysis.py:366
↓ 1 callers
Method
scale
(self, x)
Zero-shot_Learning/layers/ETSformer_EncDec.py:21
↓ 1 callers
Method
select_columns
(df)
Classification/src/datasets/data.py:195
↓ 1 callers
Function
setup
Prepare training session: read configuration from file (takes precedence), create directories. Input: args: arguments object from argparse
Classification/src/running.py:52
↓ 1 callers
Method
shift
(self, x)
Zero-shot_Learning/layers/ETSformer_EncDec.py:24
↓ 1 callers
Function
split_dataset
Splits dataset (i.e. the global datasets indices) into a test set and a training/validation set. The training/validation set is used to produ
Classification/src/datasets/datasplit.py:5
↓ 1 callers
Method
split_testset
Input: test_ratio: ratio of test set with respect to the entire dataset. Should result in an absolute number of s
Classification/src/datasets/datasplit.py:65
↓ 1 callers
Method
split_validation
Returns: train_indices: iterable of n_splits (num. of folds) numpy arrays, each array containing the global datas
Classification/src/datasets/datasplit.py:77
↓ 1 callers
Function
subsample
If a given Series is longer than `limit`, returns subsampled sequence by the specified integer factor
Classification/src/datasets/data.py:80
↓ 1 callers
Function
test
(model, test_data, test_loader, args, device, ii)
Zero-shot_Learning/utils/tools_tsf.py:280
↓ 1 callers
Function
test
(model, test_data, test_loader, args, device, itr)
Few-shot_Learning/utils/tools.py:290
↓ 1 callers
Function
test
(model, test_data, test_loader, args, device, itr)
Long-term_Forecasting/utils/tools.py:290
↓ 1 callers
Method
test
(self)
Short-term_Forecasting/exp/exp_basic.py:54
↓ 1 callers
Method
test
(self)
Anomaly_Detection/exp/exp_basic.py:54
↓ 1 callers
Method
test
(self)
Imputation/exp/exp_basic.py:54
↓ 1 callers
Method
time_delay_agg_inference
SpeedUp version of Autocorrelation (a batch-normalization style design) This is for the inference phase.
Zero-shot_Learning/layers/AutoCorrelation.py:51
↓ 1 callers
Method
time_delay_agg_training
SpeedUp version of Autocorrelation (a batch-normalization style design) This is for the training phase.
Zero-shot_Learning/layers/AutoCorrelation.py:27
↓ 1 callers
Function
time_features_from_frequency_str
Returns a list of time features that will be appropriate for the given frequency string. Parameters ---------- freq_str Frequ
Zero-shot_Learning/utils/timefeatures.py:76
↓ 1 callers
Function
time_features_from_frequency_str
Returns a list of time features that will be appropriate for the given frequency string. Parameters ---------- freq_str Frequ
Few-shot_Learning/utils/timefeatures.py:76
↓ 1 callers
Function
time_features_from_frequency_str
Returns a list of time features that will be appropriate for the given frequency string. Parameters ---------- freq_str Frequ
Long-term_Forecasting/utils/timefeatures.py:76
↓ 1 callers
Function
time_features_from_frequency_str
Returns a list of time features that will be appropriate for the given frequency string. Parameters ---------- freq_str Frequ
Short-term_Forecasting/utils/timefeatures.py:76
↓ 1 callers
Function
time_features_from_frequency_str
Returns a list of time features that will be appropriate for the given frequency string. Parameters ---------- freq_str Frequ
Anomaly_Detection/utils/timefeatures.py:76
↓ 1 callers
Function
time_features_from_frequency_str
Returns a list of time features that will be appropriate for the given frequency string. Parameters ---------- freq_str Frequ
Imputation/utils/timefeatures.py:76
↓ 1 callers
Method
topk_freq
(self, x_freq)
Zero-shot_Learning/layers/ETSformer_EncDec.py:173
↓ 1 callers
Method
train
(self, setting)
Short-term_Forecasting/exp/exp_short_term_forecasting.py:53
← previous
next →
201–300 of 1,009, ranked by callers