Skip to main content

split_train_val

anno::backends::box_embeddings_training::algorithm

Function split_train_val 

Source 
pub fn split_train_val(
    examples: &[TrainingExample],
    val_ratio: f32,
) -> (Vec<TrainingExample>, Vec<TrainingExample>)
Expand description

Split training examples into train/validation sets.

§Arguments

  • examples - All training examples
  • val_ratio - Fraction of examples to use for validation (0.0-1.0)

§Returns

(train_examples, val_examples)