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

//! Hybrid models combining different architectures.

mod convtran;
mod mlstm_fcn;
mod multi_input;
mod rnn_fcn;
mod transformer_rnn;

pub use convtran::{ConvTranPlus, ConvTranPlusConfig};
pub use mlstm_fcn::{MLSTMFCNConfig, MLSTMFCN};
pub use multi_input::{BackboneType, FusionType, MultiInputNet, MultiInputNetConfig};
pub use rnn_fcn::{RNNFCN, RNNFCNConfig, RNNFCNType};
pub use transformer_rnn::{TransformerRNNPlus, TransformerRNNPlusConfig, TransformerRNNType};