Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/GestaltCogTeam/BasicTS
/ functions
Functions
713 in github.com/GestaltCogTeam/BasicTS
⨍
Functions
713
◇
Types & classes
201
↳
Endpoints
1
↓ 1 callers
Function
load_and_preprocess_data
Load and preprocess raw data, selecting the specified channel(s).
scripts/data_preparation/Gaussian/generate_training_data.py:30
↓ 1 callers
Function
load_and_preprocess_data
Load and preprocess raw data, selecting the specified channel(s).
scripts/data_preparation/ETTm1/generate_training_data.py:37
↓ 1 callers
Function
load_and_preprocess_data
Load and preprocess raw data, selecting the specified channel(s).
scripts/data_preparation/PEMS03/generate_training_data.py:37
↓ 1 callers
Function
load_dataframe
(input_data: list)
server/engine/utils.py:41
↓ 1 callers
Function
load_input_data
(data_file)
server/web_inference.py:12
↓ 1 callers
Function
load_meta_description
Get the description of a dataset. Args: dataset_name (str): Name of the dataset. Returns: str: Description of t
src/basicts/utils/serialization.py:27
↓ 1 callers
Function
load_model
(cfg:str, ckpt:str, device:str, gpu:Optional[str],\ cnx_len:int, prd_len:int)
server/web_inference.py:19
↓ 1 callers
Method
load_runner
(self)
server/engine/engine.py:45
↓ 1 callers
Function
main
()
scripts/data_preparation/METR-LA/generate_training_data.py:128
↓ 1 callers
Function
main
()
scripts/data_preparation/ETTh1/generate_training_data.py:127
↓ 1 callers
Function
main
()
scripts/data_preparation/Illness/generate_training_data.py:127
↓ 1 callers
Function
main
()
scripts/data_preparation/ETTh2/generate_training_data.py:127
↓ 1 callers
Function
main
()
scripts/data_preparation/BeijingAirQuality/generate_training_data.py:115
↓ 1 callers
Function
main
()
scripts/data_preparation/JiNan/generate_training_data.py:127
↓ 1 callers
Function
main
()
scripts/data_preparation/Traffic/generate_training_data.py:127
↓ 1 callers
Function
main
(clean_cache=False)
scripts/data_preparation/BLAST/merge_data.py:9
↓ 1 callers
Function
main
()
scripts/data_preparation/GlobalTemp/generate_training_data.py:134
↓ 1 callers
Function
main
()
scripts/data_preparation/PEMS04/generate_training_data.py:124
↓ 1 callers
Function
main
()
scripts/data_preparation/Pulse/generate_training_data.py:86
↓ 1 callers
Function
main
()
scripts/data_preparation/GlobalWind/generate_training_data.py:135
↓ 1 callers
Function
main
()
scripts/data_preparation/ExchangeRate/generate_training_data.py:128
↓ 1 callers
Function
main
()
scripts/data_preparation/GBA/generate_training_data.py:142
↓ 1 callers
Function
main
()
scripts/data_preparation/SD/generate_training_data.py:142
↓ 1 callers
Function
main
()
scripts/data_preparation/CA/generate_training_data.py:142
↓ 1 callers
Function
main
()
scripts/data_preparation/PEMS08/generate_training_data.py:124
↓ 1 callers
Function
main
()
scripts/data_preparation/ETTm2/generate_training_data.py:127
↓ 1 callers
Function
main
()
scripts/data_preparation/Weather/generate_training_data.py:128
↓ 1 callers
Function
main
()
scripts/data_preparation/GLA/generate_training_data.py:142
↓ 1 callers
Function
main
()
scripts/data_preparation/PEMS-BAY/generate_training_data.py:128
↓ 1 callers
Function
main
()
scripts/data_preparation/PEMS07/generate_training_data.py:124
↓ 1 callers
Function
main
()
scripts/data_preparation/Electricity/generate_training_data.py:128
↓ 1 callers
Function
main
()
scripts/data_preparation/Gaussian/generate_training_data.py:86
↓ 1 callers
Function
main
()
scripts/data_preparation/ETTm1/generate_training_data.py:127
↓ 1 callers
Function
main
()
scripts/data_preparation/PEMS03/generate_training_data.py:124
↓ 1 callers
Function
main
()
examples/simplest_demo.py:6
↓ 1 callers
Function
main
()
examples/imputation/imputation_demo.py:7
↓ 1 callers
Function
main
()
examples/classification/classification_demo.py:7
↓ 1 callers
Function
main
()
examples/forecasting/forecasting_demo.py:11
↓ 1 callers
Function
masked_mse
Calculate the Masked Mean Squared Error (MSE) between the predicted and target values, while ignoring the entries in the target tensor that m
src/basicts/metrics/mse.py:4
↓ 1 callers
Method
md5
(self)
src/basicts/configs/base_config.py:159
↓ 1 callers
Method
merge
(self, shape: torch.Size, x_list: list[torch.Tensor])
src/basicts/models/MTSMixer/arch/mtsmixer_layers.py:30
↓ 1 callers
Method
noisy_top_k_gating
Noisy top-k gating. See paper: https://arxiv.org/abs/1701.06538. Args: x: input Tensor with shape [batch_size, input_size]
src/basicts/models/DUET/arch/linear_extractor_cluster.py:249
↓ 1 callers
Method
on_epoch_end
Callback at the end of each epoch to handle validation and testing. Args: epoch (int): The current epoch number.
src/basicts/runners/basicts_runner.py:662
↓ 1 callers
Method
on_epoch_start
Callback at the start of an epoch. Args: epoch (int): current epoch
src/basicts/runners/basicts_runner.py:646
↓ 1 callers
Method
on_eval_end
Callback at the end of evaluation.
src/basicts/runners/basicts_runner.py:635
↓ 1 callers
Method
on_eval_start
Callback at the start of evaluation.
src/basicts/runners/basicts_runner.py:624
↓ 1 callers
Method
on_step_end
Called after each step. Args: epoch (int): current epoch step (int): current step data (Dict): data
src/basicts/runners/basicts_runner.py:708
↓ 1 callers
Method
on_step_start
Called before each step. Args: step (int): current step
src/basicts/runners/basicts_runner.py:696
↓ 1 callers
Method
on_test_end
Callback at the end of testing.
src/basicts/runners/basicts_runner.py:607
↓ 1 callers
Method
on_test_start
Callback at the start of testing.
src/basicts/runners/basicts_runner.py:600
↓ 1 callers
Method
on_train_end
Callback at the end of the training process. Args: cfg (Dict): Configuration. train_epoch (Optional[int]): En
src/basicts/runners/basicts_runner.py:553
↓ 1 callers
Method
on_train_start
Initialize training, including support for the test data loader.
src/basicts/runners/basicts_runner.py:535
↓ 1 callers
Method
on_validate_end
Callback at the end of validating. Args: train_epoch (Optional[int]): current epoch if in training process.
src/basicts/runners/basicts_runner.py:580
↓ 1 callers
Method
on_validate_start
Callback at the start of validating. Args: train_epoch (Optional[int]): current epoch if in training process.
src/basicts/runners/basicts_runner.py:570
↓ 1 callers
Method
one_step_forward
(self, z, return_rec=False)
src/basicts/models/Koopa/arch/layers.py:77
↓ 1 callers
Function
parse_args
()
scripts/data_preparation/BLAST/merge_data.py:38
↓ 1 callers
Function
reconstruction_mask
Create a mask for the input data of a masked reconstruction task. Args: data (torch.Tensor): Input data. mask_ratio (float):
src/basicts/utils/mask.py:24
↓ 1 callers
Method
register
Add a meter to meter pool. Args: name (str): meter name. meter_type (str): meter type. fmt (str): meter ou
src/basicts/utils/meter_pool.py:16
↓ 1 callers
Method
reset
Reset all meters.
src/basicts/utils/meter_pool.py:104
↓ 1 callers
Method
reset
(self)
scripts/data_preparation/UEA/generate_training_data.py:63
↓ 1 callers
Method
save_data
Save the preprocessed data to a binary file.
scripts/data_preparation/UEA/generate_training_data.py:134
↓ 1 callers
Function
save_description
Save a description of the dataset to a JSON file.
scripts/data_preparation/METR-LA/generate_training_data.py:107
↓ 1 callers
Function
save_description
Save a description of the dataset to a JSON file.
scripts/data_preparation/ETTh1/generate_training_data.py:106
↓ 1 callers
Function
save_description
Save a description of the dataset to a JSON file.
scripts/data_preparation/Illness/generate_training_data.py:106
↓ 1 callers
Function
save_description
Save a description of the dataset to a JSON file.
scripts/data_preparation/ETTh2/generate_training_data.py:106
↓ 1 callers
Function
save_description
Save a description of the dataset to a JSON file.
scripts/data_preparation/BeijingAirQuality/generate_training_data.py:94
↓ 1 callers
Function
save_description
Save a description of the dataset to a JSON file.
scripts/data_preparation/JiNan/generate_training_data.py:106
↓ 1 callers
Function
save_description
Save a description of the dataset to a JSON file.
scripts/data_preparation/Traffic/generate_training_data.py:106
↓ 1 callers
Function
save_description
Save a description of the dataset to a JSON file.
scripts/data_preparation/GlobalTemp/generate_training_data.py:113
↓ 1 callers
Function
save_description
Save a description of the dataset to a JSON file.
scripts/data_preparation/PEMS04/generate_training_data.py:103
↓ 1 callers
Function
save_description
Save a description of the dataset to a JSON file.
scripts/data_preparation/Pulse/generate_training_data.py:65
↓ 1 callers
Function
save_description
Save a description of the dataset to a JSON file.
scripts/data_preparation/GlobalWind/generate_training_data.py:114
↓ 1 callers
Function
save_description
Save a description of the dataset to a JSON file.
scripts/data_preparation/ExchangeRate/generate_training_data.py:107
↓ 1 callers
Function
save_description
Save a description of the dataset to a JSON file.
scripts/data_preparation/GBA/generate_training_data.py:121
↓ 1 callers
Function
save_description
Save a description of the dataset to a JSON file.
scripts/data_preparation/SD/generate_training_data.py:121
↓ 1 callers
Function
save_description
Save a description of the dataset to a JSON file.
scripts/data_preparation/CA/generate_training_data.py:121
↓ 1 callers
Function
save_description
Save a description of the dataset to a JSON file.
scripts/data_preparation/PEMS08/generate_training_data.py:103
↓ 1 callers
Function
save_description
Save a description of the dataset to a JSON file.
scripts/data_preparation/ETTm2/generate_training_data.py:106
↓ 1 callers
Function
save_description
Save a description of the dataset to a JSON file.
scripts/data_preparation/Weather/generate_training_data.py:107
↓ 1 callers
Function
save_description
Save a description of the dataset to a JSON file.
scripts/data_preparation/GLA/generate_training_data.py:121
↓ 1 callers
Function
save_description
Save a description of the dataset to a JSON file.
scripts/data_preparation/PEMS-BAY/generate_training_data.py:107
↓ 1 callers
Function
save_description
Save a description of the dataset to a JSON file.
scripts/data_preparation/PEMS07/generate_training_data.py:103
↓ 1 callers
Function
save_description
Save a description of the dataset to a JSON file.
scripts/data_preparation/Electricity/generate_training_data.py:107
↓ 1 callers
Function
save_description
Save a description of the dataset to a JSON file.
scripts/data_preparation/Gaussian/generate_training_data.py:65
↓ 1 callers
Function
save_description
Save a description of the dataset to a JSON file.
scripts/data_preparation/ETTm1/generate_training_data.py:106
↓ 1 callers
Function
save_description
Save a description of the dataset to a JSON file.
scripts/data_preparation/PEMS03/generate_training_data.py:103
↓ 1 callers
Function
save_graph
Save the adjacency matrix to the output directory.
scripts/data_preparation/METR-LA/generate_training_data.py:101
↓ 1 callers
Function
save_graph
(df)
scripts/data_preparation/JiNan/generate_training_data.py:95
↓ 1 callers
Function
save_graph
(df)
scripts/data_preparation/PEMS04/generate_training_data.py:92
↓ 1 callers
Function
save_graph
Save the adjacency matrix to the output directory.
scripts/data_preparation/GBA/generate_training_data.py:108
↓ 1 callers
Function
save_graph
Save the adjacency matrix to the output directory.
scripts/data_preparation/SD/generate_training_data.py:108
↓ 1 callers
Function
save_graph
Save the adjacency matrix to the output directory.
scripts/data_preparation/CA/generate_training_data.py:108
↓ 1 callers
Function
save_graph
(df)
scripts/data_preparation/PEMS08/generate_training_data.py:92
↓ 1 callers
Function
save_graph
Save the adjacency matrix to the output directory.
scripts/data_preparation/GLA/generate_training_data.py:108
↓ 1 callers
Function
save_graph
Save the adjacency matrix to the output directory.
scripts/data_preparation/PEMS-BAY/generate_training_data.py:101
↓ 1 callers
Function
save_graph
(df)
scripts/data_preparation/PEMS07/generate_training_data.py:92
↓ 1 callers
Function
save_graph
(df)
scripts/data_preparation/PEMS03/generate_training_data.py:92
↓ 1 callers
Function
save_meta_data
Save the meta data to the output directory
scripts/data_preparation/GBA/generate_training_data.py:116
↓ 1 callers
Function
save_meta_data
Save the meta data to the output directory
scripts/data_preparation/SD/generate_training_data.py:116
↓ 1 callers
Function
save_meta_data
Save the meta data to the output directory
scripts/data_preparation/CA/generate_training_data.py:116
← previous
next →
201–300 of 713, ranked by callers