Skip to main content

CrossValidationSplit

tensorlogic_train

Trait CrossValidationSplit 

Source 
pub trait CrossValidationSplit {
    // Required methods
    fn num_splits(&self) -> usize;
    fn get_split(
        &self,
        fold: usize,
        n_samples: usize,
    ) -> TrainResult<(Vec<usize>, Vec<usize>)>;
}
Expand description

Trait for cross-validation splitting strategies.

Required Methods§

Source

fn num_splits(&self) -> usize

Get the number of splits.

Source

fn get_split( &self, fold: usize, n_samples: usize, ) -> TrainResult<(Vec<usize>, Vec<usize>)>

Get the train/validation indices for a specific fold.

§Arguments
  • fold - Fold index (0 to num_splits - 1)
  • n_samples - Total number of samples
§Returns

(train_indices, validation_indices)

Implementors§