tsai_models 0.1.1

Model zoo for tsai-rs: CNN, Transformer, ROCKET, RNN, and Tabular architectures
Documentation 
1
2
3
4
5
6
7
8
9
10
11

//! Transformer models for time series.

mod patch_tst;
mod perceiver;
mod tsit;
mod tst;

pub use patch_tst::{PatchTST, PatchTSTConfig};
pub use perceiver::{TSPerceiver, TSPerceiverConfig};
pub use tsit::{PoolingStrategy, PositionalEncodingType, TSiTPlus, TSiTPlusConfig};
pub use tst::{TSTPlus, TSTConfig};