All modules for which code is available

• spacetimeformer.callbacks
• spacetimeformer.data.csv_dataset
• spacetimeformer.data.datamodule
• spacetimeformer.data.metr_la.metr_la
• spacetimeformer.data.precip.geo_data
• spacetimeformer.data.timefeatures
• spacetimeformer.eval_stats
• spacetimeformer.forecaster
• spacetimeformer.linear_model.linear_ar
• spacetimeformer.linear_model.linear_model
• spacetimeformer.lr_scheduler.lr_scheduler
• spacetimeformer.lr_scheduler.reduce_lr_on_plateau_lr_scheduler
• spacetimeformer.lr_scheduler.transformer_lr_scheduler
• spacetimeformer.lr_scheduler.warmup_lr_scheduler
• spacetimeformer.lr_scheduler.warmup_reduce_lr_on_plateau_scheduler
• spacetimeformer.lstm_model.lstm_model
• spacetimeformer.lstnet_model.LSTNet
• spacetimeformer.lstnet_model.lstnet_model
• spacetimeformer.mtgnn_model.mtgnn_model
• spacetimeformer.plot
• spacetimeformer.spacetimeformer_model.nn.attn
• spacetimeformer.spacetimeformer_model.nn.data_dropout
• spacetimeformer.spacetimeformer_model.nn.decoder
• spacetimeformer.spacetimeformer_model.nn.embed
• spacetimeformer.spacetimeformer_model.nn.encoder
• spacetimeformer.spacetimeformer_model.nn.model
• spacetimeformer.spacetimeformer_model.nn.powernorm
• spacetimeformer.spacetimeformer_model.nn.scalenorm
• spacetimeformer.spacetimeformer_model.spacetimeformer_model
• spacetimeformer.spacetimeformer_model.utils.masking
• spacetimeformer.time2vec
• spacetimeformer.train