pub fn one_hot_encode<F: Float + Debug>(
indices: &Array1<usize>,
num_classes: usize,
) -> Result<Array2<F>>
Expand description
Calculate the one-hot encoding of a vector of indices
indices
- Vector of class indicesnum_classes
- Number of classes- A 2D array where each row is a one-hot encoded vector
use scirs2_neural::utils::one_hot_encode;
use scirs2_core::ndarray::arr1;
let indices = arr1(&[0, 2, 1]);
let one_hot = one_hot_encode::
(&indices, 3).unwrap(); assert_eq!(one_hot.shape(), &[3, 3]); assert_eq!(one_hot[[0, 0]], 1.0f64); // First sample, class 0 assert_eq!(one_hot[[1, 2]], 1.0f64); // Second sample, class 2 assert_eq!(one_hot[[2, 1]], 1.0f64); // Third sample, class 1