Struct MultiTargetSGBT 

pub struct MultiTargetSGBT<L: Loss = SquaredLoss> { /* private fields */ }

Available on crate feature alloc only.

Multi-target regression SGBT.

Wraps T independent SGBT<L> models, one per target dimension. Each model is trained and predicts independently, sharing the same configuration and loss function.

Examples

use irithyll::ensemble::multi_target::MultiTargetSGBT;
use irithyll::SGBTConfig;

let config = SGBTConfig::builder()
    .n_steps(10)
    .learning_rate(0.1)
    .grace_period(10)
    .build()
    .unwrap();

let mut model = MultiTargetSGBT::new(config, 3).unwrap();
model.train_one(&[1.0, 2.0], &[0.5, 1.0, 1.5]);
let preds = model.predict(&[1.0, 2.0]);
assert_eq!(preds.len(), 3);

Implementations

impl MultiTargetSGBT<SquaredLoss>

pub fn new(config: SGBTConfig, n_targets: usize) -> Result<Self>

Create a new multi-target SGBT with squared loss (default).

Errors

Returns IrithyllError::InvalidConfig if n_targets < 1.

impl<L: Loss + Clone> MultiTargetSGBT<L>

pub fn with_loss(config: SGBTConfig, loss: L, n_targets: usize) -> Result<Self>

Create a new multi-target SGBT with a custom loss function.

The loss is cloned for each target model.

Errors

Returns IrithyllError::InvalidConfig if n_targets < 1.

pub fn train_one(&mut self, features: &[f64], targets: &[f64])

Train on a single sample with multiple target values.

Panics

Panics if targets.len() != n_targets.

pub fn train_batch( &mut self, feature_matrix: &[Vec<f64>], target_matrix: &[Vec<f64>], )

Train on a batch of multi-target samples.

pub fn predict(&self, features: &[f64]) -> Vec<f64>

Predict all target values for a feature vector.

pub fn n_targets(&self) -> usize

Number of target dimensions.

pub fn n_samples_seen(&self) -> u64

Total samples trained.

pub fn model(&self, idx: usize) -> &SGBT<L>

Access the model for a specific target dimension.

Panics

Panics if idx >= n_targets.

pub fn models(&self) -> &[SGBT<L>]

Access all target models.

pub fn reset(&mut self)

Reset all target models.

Trait Implementations

impl<L: Loss + Clone> Clone for MultiTargetSGBT<L>

fn clone(&self) -> Self

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl<L: Debug + Loss> Debug for MultiTargetSGBT<L>

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.