From 139bbae4564347cc8b44403c89baad14647ff37f Mon Sep 17 00:00:00 2001 From: gaxler Date: Tue, 26 Jan 2021 00:01:20 -0800 Subject: [PATCH] cliipy fixes --- src/preprocessing/target_encoders.rs | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/preprocessing/target_encoders.rs b/src/preprocessing/target_encoders.rs index 81cbdbd..c282a4d 100644 --- a/src/preprocessing/target_encoders.rs +++ b/src/preprocessing/target_encoders.rs @@ -26,9 +26,9 @@ pub fn make_one_hot(label_idx: usize, num_labels: usize) -> Vec OneHotEncoder { /// Transform a slice of label types into one-hot vectors /// None is returned if unknown label is encountered pub fn transform(&self, labels: &[T]) -> Vec>> { - labels.into_iter().map(|l| self.transform_one(l)).collect() + labels.iter().map(|l| self.transform_one(l)).collect() } /// Transform a single label type into a one-hot vector @@ -130,7 +130,7 @@ impl<'a, T: Hash + Eq + Clone> OneHotEncoder { let (label_map, class_num, unique_lables) = match labels { LabelDefinition::LabelToClsNumMap(h) => { let mut _unique_lab: Vec<(T, usize)> = - h.iter().map(|(k, v)| (k.clone(), v.clone())).collect(); + h.iter().map(|(k, v)| (k.clone(), *v)).collect(); _unique_lab.sort_by(|a, b| a.1.cmp(&b.1)); let unique_lab: Vec = _unique_lab.into_iter().map(|a| a.0).collect(); (h, unique_lab.len(), unique_lab)