Struct TotoModel 

pub struct TotoModel {
    pub config: InferConfig,
    /* private fields */
}

Fields

config: InferConfig

Implementations

impl TotoModel

pub fn builder(gguf_path: impl Into<PathBuf>) -> TotoModelBuilder

Start building a TotoModel — see TotoModelBuilder.

pub fn load(gguf_path: &Path, config: InferConfig) -> Result<Self>

pub fn forecast( &self, target: &[Vec<f32>], mask: &[Vec<bool>], prediction_length: usize, ) -> Result<Vec<Vec<Vec<f32>>>>

Forecast from context.

target[n_var][ctx_len] — observed values. mask[n_var][ctx_len] — true for valid observations. Returns quantiles[9][n_var][prediction_length].

Toto is a full-context model: the prediction region is appended to the context as masked (all-zero) patches before the forward pass. The model attends over the full sequence and outputs quantiles for the future region.

Trait Implementations

impl Forecaster for TotoModel

type Config = InferConfig

fn load(gguf_path: &Path, config: InferConfig) -> Result<Self>

fn forecast( &self, context: &[Vec<f32>], mask: &[Vec<bool>], horizon: usize, ) -> Result<QuantileMatrix>

context[variate][timestep], mask[variate][timestep] (true = observed). Returns quantiles[quantile_level][variate][horizon_step].