tsai_models 0.1.2

Model zoo for tsai-rs: CNN, Transformer, ROCKET, RNN, and Tabular architectures
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21

//! Transformer models for time series.

mod gmlp;
mod patch_tst;
mod perceiver;
mod transformer;
mod ts_sequencer;
mod tsit;
mod tst;

pub use gmlp::{GMLPConfig, GMLP};
pub use patch_tst::{PatchTST, PatchTSTConfig};
pub use perceiver::{TSPerceiver, TSPerceiverConfig};
pub use transformer::{
    ActivationType, AggregationType,
    PositionalEncodingType as TransformerPosEncodingType,
    TransformerEncoderLayer, TransformerModel, TransformerModelConfig,
};
pub use ts_sequencer::{TSSequencerPlus, TSSequencerPlusConfig};
pub use tsit::{PoolingStrategy, PositionalEncodingType, TSiTPlus, TSiTPlusConfig};
pub use tst::{TSTPlus, TSTConfig};