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
Method
forward
(self, x: torch.Tensor)
src/basicts/models/MTSMixer/arch/mtsmixer_layers.py:60
Method
forward
(self, x: torch.Tensor)
src/basicts/models/MTSMixer/arch/mtsmixer_layers.py:82
Method
forward
(self,x: torch.Tensor)
src/basicts/models/MTSMixer/arch/mtsmixer_layers.py:99
Method
forward
Forward pass of the MTSMixer model. Args: inputs (torch.Tensor): Input tensor with shape [batch_size, input_len, num_fea
src/basicts/models/MTSMixer/arch/mtsmixer_arch.py:27
Method
forward
Forward function of HI. Args: inputs (torch.Tensor): shape = [B, L_in, N] Returns: torch.Tensor: model predi
src/basicts/models/HI/arch/hi_arch.py:39
Method
forward
(self, x: torch.Tensor)
src/basicts/models/DUET/arch/linear_extractor_cluster.py:133
Method
forward
Args: x: tensor shape [batch_size, input_size] loss_coef: a scalar - multiplier on load-balancing losses Ret
src/basicts/models/DUET/arch/linear_extractor_cluster.py:293
Method
forward
(self, x: torch.Tensor)
src/basicts/models/DUET/arch/mahalanobis_mask.py:57
Method
forward
Args: inputs (torch.Tensor): Input tensor with shape [batch_size, input_len, num_features]. Returns: Dict[st
src/basicts/models/DUET/arch/duet_arch.py:64
Method
forward
Forward pass of the FreTS model. Args: inputs (torch.Tensor): Input tensor of shape [batch_size, input_len, num_features
src/basicts/models/FreTS/arch/frets_arch.py:36
Method
forward
(self, x: torch.Tensor)
src/basicts/models/FreTS/arch/frets_arch.py:91
Method
forward
Forward pass of the SegRNN model. Args: inputs (torch.Tensor): [batch_size, input_len, num_features] Returns:
src/basicts/models/SegRNN/arch/segrnn_arch.py:48
Method
forward
(self, inputs)
src/basicts/models/TiDE/arch/tide_arch.py:31
Method
forward
Feed forward of TiDE. Args: inputs: Input data with shape: [batch_size, input_len, num_features] inputs_timestamps: I
src/basicts/models/TiDE/arch/tide_arch.py:87
Method
forward
( self, hidden_states: torch.Tensor, attention_mask: Optional[torch.Tensor] = None,
src/basicts/models/Informer/arch/encoder.py:25
Method
forward
(self, x: torch.Tensor)
src/basicts/models/Informer/arch/conv.py:25
Method
forward
Feed forward of Informer. Args: inputs: Input data with shape: [batch_size, input_len, num_features] targets: Future
src/basicts/models/Informer/arch/informer_arch.py:97
Method
forward
Args: inputs (Tensor): Input data with shape: [batch_size, input_len, num_features] inputs_std (Tensor): Input stand
src/basicts/models/NonstationaryTransformer/arch/ns_transformer_arch.py:77
Method
forward
Args: inputs (Tensor): Input data with shape: [batch_size, input_len, num_features] inputs_timestamps (Tensor): Inpu
src/basicts/models/NonstationaryTransformer/arch/ns_transformer_arch.py:165
Method
forward
Args: inputs (Tensor): Input data with shape: [batch_size, input_len, num_features] inputs_timestamps (Tensor): Inpu
src/basicts/models/NonstationaryTransformer/arch/ns_transformer_arch.py:214
Method
forward
Forward pass of the NonstationaryTransformerForReconstruction model. Args: inputs (Tensor): Input data with shape: [batc
src/basicts/models/NonstationaryTransformer/arch/ns_transformer_arch.py:254
Method
forward
Forward pass of the De-stationary Attention Layer. Args: hidden_states (torch.Tensor): Input tensor of shape [batch_size,
src/basicts/models/NonstationaryTransformer/arch/ns_transformer_layers.py:30
Method
forward
Forward pass of the NonstationaryTransformerEncoderLayer. Args: hidden_states (torch.Tensor): The input
src/basicts/models/NonstationaryTransformer/arch/ns_transformer_layers.py:101
Method
forward
Forward pass of the NonstationaryTransformerDecoderLayer. Args: hidden_states (torch.Tensor): The input
src/basicts/models/NonstationaryTransformer/arch/ns_transformer_layers.py:154
Method
forward
(self, x: torch.Tensor, stats: torch.Tensor)
src/basicts/models/NonstationaryTransformer/arch/ns_transformer_layers.py:239
Method
forward
Args: inputs (Tensor): Input data with shape: [batch_size, input_len, num_features] inputs_timestamps (Tensor): Inpu
src/basicts/models/iTransformer/arch/itransformer_arch.py:52
Method
forward
Forward pass of iTransformerForForecasting. Args: inputs (Tensor): Input data with shape: [batch_size, input_len, num_fe
src/basicts/models/iTransformer/arch/itransformer_arch.py:87
Method
forward
Args: inputs (Tensor): Input data with shape: [batch_size, input_len, num_features] inputs_timestamps (Tensor): Inpu
src/basicts/models/iTransformer/arch/itransformer_arch.py:129
Method
forward
Args: inputs (Tensor): Input data with shape: [batch_size, input_len, num_features] inputs_timestamps (Tensor): Inpu
src/basicts/models/iTransformer/arch/itransformer_arch.py:170
Method
forward
Feed forward of NLinear. Args: inputs (torch.Tensor): input data with shape [batch_size, input_len, num_features] Return
src/basicts/models/NLinear/arch/nlinear_arch.py:19
Method
forward
Forward pass of SOFTS model. Args: inputs (`torch.Tensor`): input tensor of shape [batch_size, seq_len, num_features]
src/basicts/models/SOFTS/arch/softs_arch.py:50
Method
forward
(self, inputs: torch.Tensor)
src/basicts/models/SOFTS/arch/star.py:19
Method
forward
(self, x: torch.Tensor)
src/basicts/models/TimeKAN/arch/timekan_layers.py:29
Method
forward
(self, level_list: List[torch.Tensor])
src/basicts/models/TimeKAN/arch/timekan_layers.py:133
Method
forward
(self, level_list: List[torch.Tensor])
src/basicts/models/TimeKAN/arch/timekan_layers.py:167
Method
forward
Forward pass of the M_KAN layer. Args: x (torch.Tensor): Input tensor of shape [batch_size, seq_len, hidden_size].
src/basicts/models/TimeKAN/arch/timekan_layers.py:195
Method
forward
Forward pass of TimeKAN model. Args: inputs (torch.Tensor): Input tensor of shape (batch_size, input_len, num_features).
src/basicts/models/TimeKAN/arch/timekan_arch.py:48
Method
from_json
Load config from a json file. Args: json_file_path (str): json file path
src/basicts/configs/base_config.py:221
Function
generate_adj_jinan
(distance_df_filename, graph_file_path, num_of_vertices)
scripts/data_preparation/JiNan/generate_adj_mx.py:65
Function
generate_adj_pems03
(distance_df_filename, graph_file_path, num_of_vertices)
scripts/data_preparation/PEMS04/generate_adj_mx.py:65
Function
generate_adj_pems03
(distance_df_filename, graph_file_path, num_of_vertices)
scripts/data_preparation/PEMS08/generate_adj_mx.py:65
Function
generate_adj_pems03
(distance_df_filename, graph_file_path, num_of_vertices)
scripts/data_preparation/PEMS07/generate_adj_mx.py:65
Function
generate_adj_pems03
(distance_df_filename, graph_file_path, num_of_vertices)
scripts/data_preparation/PEMS03/generate_adj_mx.py:65
Function
get_baseline_config_dict
()
server/engine/utils.py:29
Function
get_dataset_name
Extract the dataset name from the configuration dictionary. Args: cfg (Dict): Configuration dictionary. Returns: str: T
src/basicts/utils/config.py:4
Method
get_laplacian
(self, graph: torch.Tensor, normalize: bool)
src/basicts/models/StemGNN/arch/stemgnn_arch.py:159
Function
get_regular_settings
Get the regular settings for a dataset. Args: dataset_name (str): Name of the dataset. Returns: dict: Regular s
src/basicts/utils/serialization.py:12
Method
get_weight
Get the weight of the forward return
src/basicts/runners/taskflow/classification_taskflow.py:37
Method
get_weight
Get the weight of the forward return
src/basicts/runners/taskflow/imputation_taskflow.py:47
Method
get_weight
Get the weight of the forward return
src/basicts/runners/taskflow/forecasting_taskflow.py:46
Method
graph_fft
(self, inputs, eigenvectors)
src/basicts/models/StemGNN/arch/stemgnn_arch.py:216
Function
inference
(input_data: InputData)
server/http_server.py:21
Method
inference
The complete inference process. Args: ckpt_path (str, optional): Path to the checkpoint file. Defaults to None.
src/basicts/runners/basicts_runner.py:355
Method
inverse_transform
Reverse the Z-score normalization to recover the original data scale. This method transforms the normalized data back to its origina
src/basicts/scaler/z_score_scaler.py:98
Method
inverse_transform
Reverse the min-max normalization to recover the original data scale. This method transforms the normalized data back to its origina
src/basicts/scaler/min_max_scaler.py:91
Method
launch_evaluation
Launches the evaluation process. This method initializes the runner specified in the configuration, sets up logging, and sta
src/basicts/launcher.py:61
Function
load_adj
Load and preprocess an adjacency matrix. Args: dataset_name (str): Name of the dataset. adj_type (str): Type of adjacency ma
src/basicts/utils/serialization.py:78
Function
masked_corr
Calculate the Masked Pearson Correlation Coefficient between the predicted and target values, while ignoring the entries in the target tensor
src/basicts/metrics/corr.py:4
Function
masked_huber
Calculate the Masked Huber Loss between predicted and target values, ignoring entries in the target tensor that match the specified null valu
src/basicts/metrics/huber.py:5
Function
masked_mae
Calculate the Masked Mean Absolute Error (MAE) between the predicted and target values, while ignoring the entries in the target tensor that
src/basicts/metrics/mae.py:4
Function
masked_mape
Calculate the Masked Mean Absolute Percentage Error (MAPE) between predicted and target values, ignoring entries that are either zero or matc
src/basicts/metrics/mape.py:4
Function
masked_r2
Calculate the Masked R square between the predicted and target values, while ignoring the entries in the target tensor that match the specifi
src/basicts/metrics/r_square.py:4
Function
masked_rmse
Calculate the Masked Root Mean Squared Error (RMSE) between predicted and target values, ignoring entries in the target tensor that match the
src/basicts/metrics/rmse.py:6
Function
masked_smape
Calculate the Masked Symmetric Mean Absolute Percentage Error (SMAPE) between predicted and target values, ignoring entries that are either z
src/basicts/metrics/smape.py:5
Function
masked_wape
Calculate the Masked Weighted Absolute Percentage Error (WAPE) between predicted and target values, ignoring entries in the target tensor tha
src/basicts/metrics/wape.py:4
Method
on_backward
(self, runner, *args, **kwargs)
src/basicts/runners/callback/callback.py:39
Method
on_backward
(self, runner: "BasicTSRunner", loss: torch.Tensor)
src/basicts/runners/callback/grad_accumulation.py:29
Method
on_compute_loss
(self, runner, *args, **kwargs)
src/basicts/runners/callback/callback.py:36
Method
on_compute_loss
(self, runner: "BasicTSRunner", **kwargs)
src/basicts/runners/callback/selective_learning.py:71
Method
on_compute_loss
(self, runner: "BasicTSRunner", **kwargs)
src/basicts/runners/callback/add_aux_loss.py:22
Method
on_compute_loss
(self, runner: "BasicTSRunner", **kwargs)
src/basicts/runners/callback/curriculum_learrning.py:33
Method
on_epoch_end
(self, runner, *args, **kwargs)
src/basicts/runners/callback/callback.py:15
Method
on_epoch_end
(self, runner: "BasicTSRunner", **kwargs)
src/basicts/runners/callback/selective_learning.py:102
Method
on_epoch_start
(self, runner, *args, **kwargs)
src/basicts/runners/callback/callback.py:12
Method
on_inference_end
Callback at the end of inference.
src/basicts/runners/basicts_runner.py:747
Method
on_inference_start
Callback at the start of inference.
src/basicts/runners/basicts_runner.py:741
Method
on_optimizer_step
(self, runner, *args, **kwargs)
src/basicts/runners/callback/callback.py:42
Method
on_optimizer_step
(self, runner: "BasicTSRunner")
src/basicts/runners/callback/clip_grad.py:44
Method
on_step_end
(self, runner, *args, **kwargs)
src/basicts/runners/callback/callback.py:21
Method
on_step_start
(self, runner, *args, **kwargs)
src/basicts/runners/callback/callback.py:18
Method
on_test_end
(self, runner, *args, **kwargs)
src/basicts/runners/callback/callback.py:33
Method
on_test_start
(self, runner, *args, **kwargs)
src/basicts/runners/callback/callback.py:30
Method
on_train_begin
(self, runner)
src/basicts/models/Koopa/callback/koopa_mask_init.py:28
Method
on_train_end
(self, runner, *args, **kwargs)
src/basicts/runners/callback/callback.py:9
Method
on_train_start
(self, runner, *args, **kwargs)
src/basicts/runners/callback/callback.py:6
Method
on_train_start
(self, runner: "BasicTSRunner")
src/basicts/runners/callback/no_bp.py:15
Method
on_train_start
(self, runner: "BasicTSRunner")
src/basicts/runners/callback/grad_accumulation.py:22
Method
on_train_start
(self, runner: "BasicTSRunner")
src/basicts/runners/callback/clip_grad.py:38
Method
on_train_start
(self, runner: "BasicTSRunner")
src/basicts/runners/callback/selective_learning.py:64
Method
on_train_start
(self, runner: 'BasicTSRunner')
src/basicts/runners/callback/early_stopping.py:22
Method
on_train_start
(self, runner: "BasicTSRunner")
src/basicts/runners/callback/curriculum_learrning.py:30
Method
on_validate_end
(self, runner, *args, **kwargs)
src/basicts/runners/callback/callback.py:27
Method
on_validate_end
(self, runner: 'BasicTSRunner', train_step: int, train_epoch: Optional[int] = None)
src/basicts/runners/callback/early_stopping.py:25
Method
on_validate_start
(self, runner, *args, **kwargs)
src/basicts/runners/callback/callback.py:24
Method
plt_meters
Plot the specified type of meters in tensorboard. Args: meter_type (str): meter type. step (int): Global step value t
src/basicts/utils/meter_pool.py:85
Method
postprocess
Run the task flow
src/basicts/runners/taskflow/classification_taskflow.py:30
Method
postprocess
Run the task flow
src/basicts/runners/taskflow/imputation_taskflow.py:39
Method
postprocess
Run the task flow
src/basicts/runners/taskflow/forecasting_taskflow.py:36
Method
preprocess
Run the task flow
src/basicts/runners/taskflow/classification_taskflow.py:16
Method
preprocess
Run the task flow
src/basicts/runners/taskflow/imputation_taskflow.py:16
← previous
next →
601–700 of 713, ranked by callers