Functions — github.com/GestaltCogTeam/BasicTS

hub / github.com/GestaltCogTeam/BasicTS / functions

Functions713 in github.com/GestaltCogTeam/BasicTS

⨍ Functions 713◇ Types & classes 201↳ Endpoints 1

↓ 1 callersFunctionsave_meta_data

Save the meta data to the output directory

scripts/data_preparation/GLA/generate_training_data.py:116

↓ 1 callersMethodself_attn_forward

Forward pass of the self-attention layer. Args: hidden_states (torch.Tensor): The input hidden states.

src/basicts/modules/transformer/decoder.py:43

↓ 1 callersMethodself_attn_forward

Forward pass of the self-attention layer. Args: hidden_states (torch.Tensor): The input hidden states.

src/basicts/modules/transformer/decoder.py:267

↓ 1 callersMethodself_attn_forward

Forward pass of the self-attention layer. Args: hidden_states (torch.Tensor): The input hidden states.

src/basicts/modules/transformer/decoder.py:437

↓ 1 callersMethodself_graph_attention

(self, inputs: torch.Tensor)

src/basicts/models/StemGNN/arch/stemgnn_arch.py:201

↓ 1 callersMethodset_env

Setup runtime env, include tf32, seed and determinacy. env config template: ``` cfg.tf32 = False cfg.seed = 42

src/basicts/runners/basicts_runner.py:838

↓ 1 callersMethodsnorm

Normalization over the spatial dimension. Args: x (torch.Tensor): Input tensor with shape [batch_size, hidden_size, num_

src/basicts/modules/norm/stnorm.py:38

↓ 1 callersMethodspe_seq_cell

(self, inputs: torch.Tensor)

src/basicts/models/StemGNN/arch/stemgnn_arch.py:67

↓ 1 callersFunctionsplit_and_save_data

Save the preprocessed data to a binary file.

scripts/data_preparation/METR-LA/generate_training_data.py:71

↓ 1 callersFunctionsplit_and_save_data

Save the preprocessed data to a binary file.

scripts/data_preparation/ETTh1/generate_training_data.py:76

↓ 1 callersFunctionsplit_and_save_data

Save the preprocessed data to a binary file.

scripts/data_preparation/Illness/generate_training_data.py:76

↓ 1 callersFunctionsplit_and_save_data

Save the preprocessed data to a binary file.

scripts/data_preparation/ETTh2/generate_training_data.py:76

↓ 1 callersFunctionsplit_and_save_data

Save the preprocessed data to a binary file.

scripts/data_preparation/BeijingAirQuality/generate_training_data.py:63

↓ 1 callersFunctionsplit_and_save_data

Save the preprocessed data to a binary file.

scripts/data_preparation/JiNan/generate_training_data.py:65

↓ 1 callersFunctionsplit_and_save_data

Save the preprocessed data to a binary file.

scripts/data_preparation/Traffic/generate_training_data.py:76

↓ 1 callersFunctionsplit_and_save_data

Save the preprocessed data to a binary file.

scripts/data_preparation/GlobalTemp/generate_training_data.py:83

↓ 1 callersFunctionsplit_and_save_data

Save the preprocessed data to a binary file.

scripts/data_preparation/PEMS04/generate_training_data.py:62

↓ 1 callersFunctionsplit_and_save_data

Save the preprocessed data to a binary file.

scripts/data_preparation/Pulse/generate_training_data.py:42

↓ 1 callersFunctionsplit_and_save_data

Save the preprocessed data to a binary file.

scripts/data_preparation/GlobalWind/generate_training_data.py:84

↓ 1 callersFunctionsplit_and_save_data

Save the preprocessed data to a binary file.

scripts/data_preparation/ExchangeRate/generate_training_data.py:77

↓ 1 callersFunctionsplit_and_save_data

Save the preprocessed data to a binary file.

scripts/data_preparation/GBA/generate_training_data.py:78

↓ 1 callersFunctionsplit_and_save_data

Save the preprocessed data to a binary file.

scripts/data_preparation/SD/generate_training_data.py:78

↓ 1 callersFunctionsplit_and_save_data

Save the preprocessed data to a binary file.

scripts/data_preparation/CA/generate_training_data.py:78

↓ 1 callersFunctionsplit_and_save_data

Save the preprocessed data to a binary file.

scripts/data_preparation/PEMS08/generate_training_data.py:62

↓ 1 callersFunctionsplit_and_save_data

Save the preprocessed data to a binary file.

scripts/data_preparation/ETTm2/generate_training_data.py:76

↓ 1 callersFunctionsplit_and_save_data

Save the preprocessed data to a binary file.

scripts/data_preparation/Weather/generate_training_data.py:77

↓ 1 callersFunctionsplit_and_save_data

Save the preprocessed data to a binary file.

scripts/data_preparation/GLA/generate_training_data.py:78

↓ 1 callersFunctionsplit_and_save_data

Save the preprocessed data to a binary file.

scripts/data_preparation/PEMS-BAY/generate_training_data.py:71

↓ 1 callersFunctionsplit_and_save_data

Save the preprocessed data to a binary file.

scripts/data_preparation/PEMS07/generate_training_data.py:62

↓ 1 callersFunctionsplit_and_save_data

Save the preprocessed data to a binary file.

scripts/data_preparation/Electricity/generate_training_data.py:77

↓ 1 callersFunctionsplit_and_save_data

Save the preprocessed data to a binary file.

scripts/data_preparation/Gaussian/generate_training_data.py:42

↓ 1 callersFunctionsplit_and_save_data

Save the preprocessed data to a binary file.

scripts/data_preparation/ETTm1/generate_training_data.py:76

↓ 1 callersFunctionsplit_and_save_data

Save the preprocessed data to a binary file.

scripts/data_preparation/PEMS03/generate_training_data.py:62

↓ 1 callersFunctiontest_crossformer_smoke_test

()

tests/smoke_test/test_crossformer.py:13

↓ 1 callersFunctiontest_dlinear_smoke_test

()

tests/smoke_test/test_dlinear.py:14

↓ 1 callersFunctiontest_duet_smoke_test

()

tests/smoke_test/test_duet.py:14

↓ 1 callersFunctiontest_fits_smoke_test

()

tests/smoke_test/test_fits.py:13

↓ 1 callersFunctiontest_frets_smoke_test

()

tests/smoke_test/test_frets.py:13

↓ 1 callersFunctiontest_informer_smoke_test

()

tests/smoke_test/test_informer.py:14

↓ 1 callersFunctiontest_itransformerforc_smoke_test

()

tests/smoke_test/test_itransformerforc.py:15

↓ 1 callersFunctiontest_itransformerforr_smoke_test

()

tests/smoke_test/test_itransformerforr.py:15

↓ 1 callersFunctiontest_koopa_smoke_test

()

tests/smoke_test/test_koopa.py:14

↓ 1 callersFunctiontest_leddam_smoke_test

()

tests/smoke_test/test_leddam.py:13

↓ 1 callersFunctiontest_lightts_smoke_test

()

tests/smoke_test/test_lightts.py:13

↓ 1 callersFunctiontest_mtsmixer_smoke_test

()

tests/smoke_test/test_mtsmixer.py:13

↓ 1 callersFunctiontest_nlinear_smoke_test

()

tests/smoke_test/test_nlinear.py:13

↓ 1 callersFunctiontest_patchtst_smoke_test

()

tests/smoke_test/test_patchtstforf.py:13

↓ 1 callersFunctiontest_patchtstforr_smoke_test

()

tests/smoke_test/test_patchtstforr.py:14

↓ 1 callersFunctiontest_segrnn_smoke_test

()

tests/smoke_test/test_segrnn.py:13

↓ 1 callersFunctiontest_sparsetsf_smoke_test

()

tests/smoke_test/test_sparsetsf.py:13

↓ 1 callersFunctiontest_stemgnn_smoke_test

()

tests/smoke_test/test_stemgnn.py:13

↓ 1 callersFunctiontest_stid_smoke_test

()

tests/smoke_test/test_stid.py:13

↓ 1 callersFunctiontest_timekan_smoke_test

()

tests/smoke_test/test_timekan.py:13

↓ 1 callersFunctiontest_timesnet_smoke_test

()

tests/smoke_test/test_timesnet.py:13

↓ 1 callersMethodtnorm

Temporal normalization. Args: x (torch.Tensor): Input tensor with shape [batch_size, hidden_size, num_features].

src/basicts/modules/norm/stnorm.py:54

Method__getattr__

(self, name)

src/basicts/runners/callback/selective_learning.py:213

Method__getitem__

(self, key)

src/basicts/configs/base_config.py:210

Method__getitem__

(self, layer_idx: int)

src/basicts/modules/transformer/kv_cache.py:22

Method__getitem__

Retrieves a sample from the dataset at the specified index, considering both the input and output lengths. Args: index (

src/basicts/data/tsi_dataset.py:68

Method__getitem__

(self, idx: int)

src/basicts/data/blast.py:211

Method__getitem__

Retrieves a sample from the dataset at the specified index, considering both the input and output lengths. Args: index (

src/basicts/data/uea_dataset.py:69

Method__getitem__

Retrieves a sample from the dataset at the specified index, considering both the input and output lengths. Args: index (

src/basicts/data/tsf_dataset.py:71

Method__getstate__

(self)

src/basicts/data/blast.py:248

Method__init__

(self, **kwargs)

src/basicts/configs/base_config.py:170

Method__init__

(self, kernel_size: int, stride: int = 1)

src/basicts/modules/decomposition.py:45

Method__init__

(self, kernel_size: Sequence[int], stride: int = 1)

src/basicts/modules/decomposition.py:69

Method__init__

(self, top_k: int = 5)

src/basicts/modules/decomposition.py:100

Method__init__

Initialize the MLP layer. Args: hidden_size (int): Input hidden dimension. intermediate_size (int): Intermed

src/basicts/modules/mlps.py:53

Method__init__

Args: hidden_size (int): size of hidden dimension. max_len (int, optional): maximum length of the time series. Defaul

src/basicts/modules/embed/tst_embed.py:14

Method__init__

( self, hidden_size: int, timestamp_sizes: Sequence[int])

src/basicts/modules/embed/tst_embed.py:68

Method__init__

( self, num_features: int, hidden_size: int, embed_type: Liter

src/basicts/modules/embed/tst_embed.py:101

Method__init__

( self, seq_len: int, hidden_size: int, dropout: float = 0.1)

src/basicts/modules/embed/tst_embed.py:150

Method__init__

( self, hidden_size: int, patch_len: int = 16, stride: int = 8

src/basicts/modules/embed/tst_embed.py:188

Method__init__

:param num_features: the number of features or channels :param eps: a value added for numerical stability :param affine: if T

src/basicts/modules/norm/revin.py:17

Method__init__

(self, hidden_size: int)

src/basicts/modules/norm/layer_norm.py:10

Method__init__

( self, hidden_size: int, enable_snorm: bool = True, enable_tn

src/basicts/modules/norm/stnorm.py:13

Method__init__

(self, hidden_size: torch.Tensor, eps: float = 1e-6)

src/basicts/modules/norm/rmsnorm.py:11

Method__init__

Args: self_attn (nn.Module): The self-attention layer. ffn_layer (nn.Module): The fee

src/basicts/modules/transformer/decoder.py:20

Method__init__

Args: decoder_layers (nn.ModuleList): The decoder layers. layer_norm (Optional[nn.Module], optional):

src/basicts/modules/transformer/decoder.py:164

Method__init__

( self, self_attn: nn.Module, cross_attn: nn.Module, ffn_layer: nn.Module,

src/basicts/modules/transformer/decoder.py:252

Method__init__

( self, self_attn: nn.Module, cross_attn: nn.Module, ffn_layer: nn.Module,

src/basicts/modules/transformer/decoder.py:422

Method__init__

Args: decoder_layers (nn.ModuleList): The decoder layers. layer_norm (Optional[nn.Module], optional):

src/basicts/modules/transformer/decoder.py:705

Method__init__

( self, dim: int, max_position_embeddings: int = 2048, base: f

src/basicts/modules/transformer/rope.py:18

Method__init__

(self, _distributed_cache_data: Optional[Iterable] = None)

src/basicts/modules/transformer/kv_cache.py:13

Method__init__

( self, self_attn: nn.Module, ffn_layer: nn.Module, layer_norm: Union[Callable

src/basicts/modules/transformer/encoder.py:17

Method__init__

(self, hidden_size: int, n_heads: int, dropout: float = 0.0

src/basicts/modules/transformer/attentions/multi_head_attention.py:21

Method__init__

(self, hidden_size: int, n_heads: int, factor: int = 5,

src/basicts/modules/transformer/attentions/prob_attention.py:14

Method__init__

(self, hidden_size: int, n_heads: int, dropout: float = 0.0, factor: float = 1.0)

src/basicts/modules/transformer/attentions/auto_correlation.py:16

Method__init__

(self, dataloader)

src/basicts/utils/dataset.py:10

Method__init__

(self)

src/basicts/utils/meter_pool.py:13

Method__init__

Initialize the BaseRunner. - Initialize logger - Set environment variables (number of gpus, seeds, cuda related setti

src/basicts/runners/basicts_runner.py:63

Method__init__

(self, optimizer: Optimizer, num_warmup_steps: int, num_training_steps: int, num_cycles: int = 1, last_epoch:

src/basicts/runners/optim/lr_schedulers.py:76

Method__init__

(self, params, lr: Union[float, Tensor] = 1e-3, betas: Tuple[float, float] = (0.9, 0.9

src/basicts/runners/optim/optimizers.py:17

Method__init__

(self, callbacks: list[BasicTSCallback] = None)

src/basicts/runners/callback/callback.py:51

Method__init__

(self, accumulation_steps: int)

src/basicts/runners/callback/grad_accumulation.py:17

Method__init__

(self, max_norm: float, norm_type: float = 2, error_if_nonf

src/basicts/runners/callback/clip_grad.py:25

Method__init__

( self, r_u: Optional[float] = None, r_a: Optional[float] = None,

src/basicts/runners/callback/selective_learning.py:36

Method__init__

(self, dataloader: DataLoader)

src/basicts/runners/callback/selective_learning.py:184

Method__init__

Args: losses: Iterable[str], keys of losses in `forward_return` that will be added. Default is ["aux_loss"].

src/basicts/runners/callback/add_aux_loss.py:14

Method__init__

(self, patience: int = 10)

src/basicts/runners/callback/early_stopping.py:18

← previous next →301–400 of 713, ranked by callers