pub struct NeuralDiffusionSchemaConfig {
pub hidden_dims: Vec<usize>,
pub timestep_embed_dim: usize,
pub learning_rate: f64,
pub training_epochs: usize,
pub batch_size: usize,
pub hybrid_weight: f64,
pub hybrid_strategy: String,
pub neural_columns: Vec<String>,
pub checkpoint_path: Option<String>,
}Expand description
Neural diffusion backend configuration.
Controls the candle-based neural score network that learns joint distributions
from training data for the neural and hybrid diffusion backends.
Fields§
Hidden layer dimensions for the score network MLP.
timestep_embed_dim: usizeDimensionality of the timestep embedding.
learning_rate: f64Learning rate for training.
training_epochs: usizeNumber of training epochs.
batch_size: usizeTraining batch size.
hybrid_weight: f64Blend weight for hybrid mode (0.0 = all statistical, 1.0 = all neural).
hybrid_strategy: StringHybrid blending strategy: “weighted_average”, “column_select”, “threshold”.
neural_columns: Vec<String>Columns to apply neural generation to (empty = all numeric columns).
checkpoint_path: Option<String>v4.4.0+ Optional path to a pre-trained score-network checkpoint
(.safetensors). When set, the orchestrator loads the
checkpoint instead of training from the first batch — useful
for long-running production deployments where training cost
dominates per-run cost. When empty, the orchestrator trains
on the first generated JE amounts.
Trait Implementations§
Source§impl Clone for NeuralDiffusionSchemaConfig
impl Clone for NeuralDiffusionSchemaConfig
Source§fn clone(&self) -> NeuralDiffusionSchemaConfig
fn clone(&self) -> NeuralDiffusionSchemaConfig
1.0.0 (const: unstable) · Source§fn clone_from(&mut self, source: &Self)
fn clone_from(&mut self, source: &Self)
source. Read more