work/smartcore
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

style fixes

Browse Source
This commit is contained in:
gaxler
2021-01-30 20:09:52 -08:00
parent ca0816db97
commit 863be5ef75
1 changed files with 10 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/preprocessing/categorical_encoder.rs
+10 -10
View File
@@ -262,9 +262,9 @@ mod tests {
} }
#[test] #[test]
fn test_fit() { fn test_fit() {
let (X, _) = build_fake_matrix(); let (x, _) = build_fake_matrix();
let params = OneHotEncoderParams::from_cat_idx(&[1, 3]); let params = OneHotEncoderParams::from_cat_idx(&[1, 3]);
let oh_enc = OneHotEncoder::fit(&X, params).unwrap(); let oh_enc = OneHotEncoder::fit(&x, params).unwrap();
assert_eq!(oh_enc.series_encoders.len(), 2); assert_eq!(oh_enc.series_encoders.len(), 2);
let num_cat: Vec<usize> = oh_enc let num_cat: Vec<usize> = oh_enc
@@ -277,17 +277,17 @@ mod tests {
#[test] #[test]
fn matrix_transform_test() { fn matrix_transform_test() {
let (X, expectedX) = build_fake_matrix(); let (x, expected_x) = build_fake_matrix();
let params = OneHotEncoderParams::from_cat_idx(&[1, 3]); let params = OneHotEncoderParams::from_cat_idx(&[1, 3]);
let oh_enc = OneHotEncoder::fit(&X, params).unwrap(); let oh_enc = OneHotEncoder::fit(&x, params).unwrap();
let nm = oh_enc.transform(&X).unwrap(); let nm = oh_enc.transform(&x).unwrap();
assert_eq!(nm, expectedX); assert_eq!(nm, expected_x);
let (X, expectedX) = build_cat_first_and_last(); let (x, expected_x) = build_cat_first_and_last();
let params = OneHotEncoderParams::from_cat_idx(&[0, 2]); let params = OneHotEncoderParams::from_cat_idx(&[0, 2]);
let oh_enc = OneHotEncoder::fit(&X, params).unwrap(); let oh_enc = OneHotEncoder::fit(&x, params).unwrap();
let nm = oh_enc.transform(&X).unwrap(); let nm = oh_enc.transform(&x).unwrap();
assert_eq!(nm, expectedX); assert_eq!(nm, expected_x);
} }
#[test] #[test]