diff --git a/src/preprocessing/target_encoders.rs b/src/preprocessing/target_encoders.rs index c282a4d..44a5c05 100644 --- a/src/preprocessing/target_encoders.rs +++ b/src/preprocessing/target_encoders.rs @@ -24,13 +24,7 @@ enum LabelDefinition { pub fn make_one_hot(label_idx: usize, num_labels: usize) -> Vec { let (pos, neg) = (T::from_f64(1f64).unwrap(), T::from_f64(0f64).unwrap()); (0..num_labels) - .map(|idx| { - if idx == label_idx { - pos - } else { - neg - } - }) + .map(|idx| if idx == label_idx { pos } else { neg }) .collect() }