Skip to main content

train_val_split

thrust_rl::buffer::rollout

Function train_val_split 

Source 
pub fn train_val_split(
    buffer: &RolloutBuffer,
    train_ratio: f32,
) -> (Vec<usize>, Vec<usize>)
Expand description

Split buffer into train/validation sets

Useful for hyperparameter tuning or model validation.

§Arguments

  • buffer - Source buffer to split
  • train_ratio - Fraction of data to use for training (0.0 to 1.0)

§Returns

(train_indices, val_indices)