dsipts.models package

Subpackages

• dsipts.models.autoformer package
  • Submodules
    • dsipts.models.autoformer.layers module
      • AutoCorrelation
      • AutoCorrelationLayer
      • my_Layernorm
      • moving_avg
      • series_decomp
      • EncoderLayer
      • Encoder
      • DecoderLayer
      • Decoder
      • PositionalEmbedding
• dsipts.models.crossformer package
  • Submodules
    • dsipts.models.crossformer.attn module
      • FullAttention
      • AttentionLayer
      • TwoStageAttentionLayer
    • dsipts.models.crossformer.cross_decoder module
      • DecoderLayer
      • Decoder
    • dsipts.models.crossformer.cross_embed module
      • DSW_embedding
    • dsipts.models.crossformer.cross_encoder module
      • SegMerging
      • scale_block
      • Encoder
• dsipts.models.d3vae package
  • Submodules
    • dsipts.models.d3vae.diffusion_process module
      • get_beta_schedule()
      • default()
      • extract()
      • noise_like()
      • GaussianDiffusion
    • dsipts.models.d3vae.embedding module
      • PositionalEmbedding
      • TokenEmbedding
      • TemporalEmbedding
      • DataEmbedding
    • dsipts.models.d3vae.encoder module
      • Cell
      • soft_clamp5()
      • sample_normal_jit()
      • Normal
      • NormalDecoder
      • log_density_gaussian()
      • Encoder
    • dsipts.models.d3vae.model module
      • diffusion_generate
      • denoise_net
      • pred_net
      • Discriminator
    • dsipts.models.d3vae.neural_operations module
      • SyncBatchNormSwish
      • get_skip_connection()
      • norm()
      • logit()
      • act()
      • SwishFN
      • Swish
      • normalize_weight_jit()
      • Conv2D
      • Identity
      • SyncBatchNorm
      • get_batchnorm()
      • ELUConv
      • BNELUConv
      • BNSwishConv
      • FactorizedReduce
      • UpSample
      • EncCombinerCell
      • DecCombinerCell
      • ConvBNSwish
      • SE
      • InvertedResidual
    • dsipts.models.d3vae.resnet module
      • weights_init()
      • MyConvo2d
      • Square
      • Swish
      • MeanPoolConv
      • ConvMeanPool
      • ResidualBlock
      • Res12_Quadratic
    • dsipts.models.d3vae.utils module
      • average_tensor()
      • get_stride_for_cell_type()
      • get_input_size()
      • groups_per_scale()
      • get_arch_cells()
• dsipts.models.informer package
  • Submodules
    • dsipts.models.informer.attn module
      • TriangularCausalMask
      • ProbMask
      • FullAttention
      • ProbAttention
      • AttentionLayer
    • dsipts.models.informer.decoder module
      • DecoderLayer
      • Decoder
    • dsipts.models.informer.embed module
      • PositionalEmbedding
      • TokenEmbedding
      • FixedEmbedding
      • TemporalEmbedding
      • TimeFeatureEmbedding
      • DataEmbedding
    • dsipts.models.informer.encoder module
      • ConvLayer
      • EncoderLayer
      • Encoder
      • EncoderStack
• dsipts.models.itransformer package
  • Submodules
    • dsipts.models.itransformer.Embed module
      • PositionalEmbedding
      • TokenEmbedding
      • FixedEmbedding
      • TemporalEmbedding
      • TimeFeatureEmbedding
      • DataEmbedding
      • DataEmbedding_inverted
    • dsipts.models.itransformer.SelfAttention_Family module
      • TriangularCausalMask
      • ProbMask
      • FlowAttention
      • FlashAttention
      • FullAttention
      • ProbAttention
      • AttentionLayer
    • dsipts.models.itransformer.Transformer_EncDec module
      • ConvLayer
      • EncoderLayer
      • Encoder
      • DecoderLayer
      • Decoder
• dsipts.models.patchtst package
  • Submodules
    • dsipts.models.patchtst.layers module
      • Transpose
      • moving_avg
      • series_decomp
      • PositionalEncoding()
      • SinCosPosEncoding()
      • Coord2dPosEncoding()
      • Coord1dPosEncoding()
      • positional_encoding()
      • PatchTST_backbone
      • Flatten_Head
      • TSTiEncoder
      • TSTEncoder
      • TSTEncoderLayer
      • RevIN
• dsipts.models.samformer package
  • Submodules
    • dsipts.models.samformer.utils module
      • scaled_dot_product_attention()
      • RevIN
      • SAM
• dsipts.models.tft package
  • Submodules
    • dsipts.models.tft.sub_nn module
      • embedding_cat_variables
      • LSTM_Model
      • GLU
      • GRN
      • ResidualConnection
      • InterpretableMultiHead
• dsipts.models.vva package
  • Submodules
    • dsipts.models.vva.minigpt module
      • NewGELU
      • CausalSelfAttention
      • Block
    • dsipts.models.vva.vqvae module
      • VectorQuantizer
      • VectorQuantizerEMA
      • Residual
      • ResidualStack
      • Encoder
      • Jitter
      • Decoder
      • VQVAE
• dsipts.models.xlstm package
  • Submodules
    • dsipts.models.xlstm.xLSTM module
      • mLSTM
      • sLSTM
      • xLSTMBlock
      • xLSTM

Submodules

• dsipts.models.Autoformer module
  • Autoformer
    • Autoformer.handle_multivariate
    • Autoformer.handle_future_covariates
    • Autoformer.handle_categorical_variables
    • Autoformer.handle_quantile_loss
    • Autoformer.description
    • Autoformer.__init__()
    • Autoformer.forward()
• dsipts.models.CrossFormer module
  • CrossFormer
    • CrossFormer.handle_multivariate
    • CrossFormer.handle_future_covariates
    • CrossFormer.handle_categorical_variables
    • CrossFormer.handle_quantile_loss
    • CrossFormer.description
    • CrossFormer.__init__()
    • CrossFormer.forward()
• dsipts.models.D3VAE module
  • copy_parameters()
  • D3VAE
    • D3VAE.__init__()
    • D3VAE.forward()
    • D3VAE.inference()
• dsipts.models.Diffusion module
  • Diffusion
    • Diffusion.handle_multivariate
    • Diffusion.handle_future_covariates
    • Diffusion.handle_categorical_variables
    • Diffusion.handle_quantile_loss
    • Diffusion.description
    • Diffusion.__init__()
    • Diffusion.forward()
    • Diffusion.inference()
    • Diffusion.cat_categorical_vars()
    • Diffusion.remove_var()
    • Diffusion.improving_weight_during_training()
    • Diffusion.q_sample()
    • Diffusion.normal_kl()
    • Diffusion.gaussian_likelihood()
    • Diffusion.gaussian_log_likelihood()
  • SubNet1
    • SubNet1.__init__()
    • SubNet1.forward()
  • SubNet2
    • SubNet2.__init__()
    • SubNet2.forward()
  • SubNet3
    • SubNet3.__init__()
    • SubNet3.forward()
• dsipts.models.DilatedConv module
  • GLU
    • GLU.__init__()
    • GLU.forward()
  • Block
    • Block.__init__()
    • Block.forward()
  • DilatedConv
    • DilatedConv.handle_multivariate
    • DilatedConv.handle_future_covariates
    • DilatedConv.handle_categorical_variables
    • DilatedConv.handle_quantile_loss
    • DilatedConv.description
    • DilatedConv.__init__()
    • DilatedConv.forward()
    • DilatedConv.inference()
• dsipts.models.DilatedConvED module
  • GLU
    • GLU.__init__()
    • GLU.forward()
  • Block
    • Block.__init__()
    • Block.forward()
  • DilatedConvED
    • DilatedConvED.handle_multivariate
    • DilatedConvED.handle_future_covariates
    • DilatedConvED.handle_categorical_variables
    • DilatedConvED.handle_quantile_loss
    • DilatedConvED.description
    • DilatedConvED.__init__()
    • DilatedConvED.forward()
• dsipts.models.ITransformer module
  • ITransformer
    • ITransformer.handle_multivariate
    • ITransformer.handle_future_covariates
    • ITransformer.handle_categorical_variables
    • ITransformer.handle_quantile_loss
    • ITransformer.description
    • ITransformer.__init__()
    • ITransformer.forecast()
    • ITransformer.forward()
• dsipts.models.Informer module
  • Informer
    • Informer.handle_multivariate
    • Informer.handle_future_covariates
    • Informer.handle_categorical_variables
    • Informer.handle_quantile_loss
    • Informer.description
    • Informer.__init__()
    • Informer.forward()
• dsipts.models.LinearTS module
  • moving_avg
    • moving_avg.__init__()
    • moving_avg.forward()
  • series_decomp
    • series_decomp.__init__()
    • series_decomp.forward()
  • LinearTS
    • LinearTS.handle_multivariate
    • LinearTS.handle_future_covariates
    • LinearTS.handle_categorical_variables
    • LinearTS.handle_quantile_loss
    • LinearTS.description
    • LinearTS.__init__()
    • LinearTS.forward()
• dsipts.models.PatchTST module
  • PatchTST
    • PatchTST.handle_multivariate
    • PatchTST.handle_future_covariates
    • PatchTST.handle_categorical_variables
    • PatchTST.handle_quantile_loss
    • PatchTST.description
    • PatchTST.__init__()
    • PatchTST.forward()
• dsipts.models.Persistent module
  • Persistent
    • Persistent.handle_multivariate
    • Persistent.handle_future_covariates
    • Persistent.handle_categorical_variables
    • Persistent.handle_quantile_loss
    • Persistent.description
    • Persistent.__init__()
    • Persistent.forward()
• dsipts.models.RNN module
  • MyBN
    • MyBN.__init__()
    • MyBN.forward()
  • RNN
    • RNN.handle_multivariate
    • RNN.handle_future_covariates
    • RNN.handle_categorical_variables
    • RNN.handle_quantile_loss
    • RNN.__init__()
    • RNN.forward()
• dsipts.models.Samformer module
  • Samformer
    • Samformer.handle_multivariate
    • Samformer.handle_future_covariates
    • Samformer.handle_categorical_variables
    • Samformer.handle_quantile_loss
    • Samformer.description
    • Samformer.__init__()
    • Samformer.forward()
• dsipts.models.TFT module
  • TFT
    • TFT.handle_multivariate
    • TFT.handle_future_covariates
    • TFT.handle_categorical_variables
    • TFT.handle_quantile_loss
    • TFT.description
    • TFT.__init__()
    • TFT.forward()
    • TFT.remove_var()
• dsipts.models.TIDE module
  • TIDE
    • TIDE.handle_multivariate
    • TIDE.handle_future_covariates
    • TIDE.handle_categorical_variables
    • TIDE.handle_quantile_loss
    • TIDE.description
    • TIDE.__init__()
    • TIDE.forward()
    • TIDE.cat_categorical_vars()
    • TIDE.remove_var()
  • ResidualBlock
    • ResidualBlock.__init__()
    • ResidualBlock.forward()
• dsipts.models.VQVAEA module
  • random()
  • VQVAEA
    • VQVAEA.handle_multivariate
    • VQVAEA.handle_future_covariates
    • VQVAEA.handle_categorical_variables
    • VQVAEA.handle_quantile_loss
    • VQVAEA.description
    • VQVAEA.__init__()
    • VQVAEA.gpt()
    • VQVAEA.forward()
    • VQVAEA.generate()
    • VQVAEA.inference()
• dsipts.models.VVA module
  • VVA
    • VVA.handle_multivariate
    • VVA.handle_future_covariates
    • VVA.handle_categorical_variables
    • VVA.handle_quantile_loss
    • VVA.description
    • VVA.__init__()
    • VVA.configure_optimizers()
    • VVA.forward()
    • VVA.generate()
    • VVA.inference()
• dsipts.models.base module
  • standardize_momentum()
  • dilate_loss()
  • Base
    • Base.handle_multivariate
    • Base.handle_future_covariates
    • Base.handle_categorical_variables
    • Base.handle_quantile_loss
    • Base.description
    • Base.__init__()
    • Base.forward()
    • Base.inference()
• dsipts.models.utils module
  • get_scope()
  • SinkhornDistance
    • SinkhornDistance.__init__()
    • SinkhornDistance.compute()
    • SinkhornDistance.M()
    • SinkhornDistance.ave()
  • QuantileLossMO
    • QuantileLossMO.__init__()
    • QuantileLossMO.forward()
  • L1Loss
    • L1Loss.__init__()
    • L1Loss.forward()
  • Permute
    • Permute.__init__()
    • Permute.forward()
  • get_activation()
  • weight_init_zeros()
  • weight_init()
  • pairwise_distances()
  • compute_softdtw()
  • compute_softdtw_backward()
  • SoftDTWBatch
    • SoftDTWBatch.forward()
    • SoftDTWBatch.backward()
  • my_max()
  • my_min()
  • my_max_hessian_product()
  • my_min_hessian_product()
  • dtw_grad()
  • dtw_hessian_prod()
  • PathDTWBatch
    • PathDTWBatch.forward()
    • PathDTWBatch.backward()