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

//! ROCKET family models.

mod hydra;
mod hydra_multirocket;
mod minirocket;
mod multirocket;
mod rocket;

pub use hydra::{HydraPlus, HydraPlusConfig, HydraPooling};
pub use hydra_multirocket::{HydraMultiRocketPlus, HydraMultiRocketPlusConfig};
pub use minirocket::{MiniRocket, MiniRocketConfig, MiniRocketFeatures, KERNEL_PATTERNS};
pub use multirocket::{FeatureType, MultiRocket, MultiRocketConfig, MultiRocketFeatures};
pub use rocket::{RandomKernel, Rocket, RocketConfig, RocketFeatures};