From 78e53a28e7aa67df7300f9713dabe8b3271c9d87 Mon Sep 17 00:00:00 2001 From: "Lorenzo (Mec-iS)" Date: Mon, 31 Oct 2022 19:28:24 +0000 Subject: [PATCH] apply fmt --- src/ensemble/random_forest_classifier.rs | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/src/ensemble/random_forest_classifier.rs b/src/ensemble/random_forest_classifier.rs index 749f4fd..44bd4e3 100644 --- a/src/ensemble/random_forest_classifier.rs +++ b/src/ensemble/random_forest_classifier.rs @@ -820,9 +820,7 @@ mod tests { &[6.6, 2.9, 4.6, 1.3], &[5.2, 2.7, 3.9, 1.4], ]); - let y = vec![ - 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, - ]; + let y = vec![0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1]; let classifier = RandomForestClassifier::fit( &x, @@ -832,7 +830,7 @@ mod tests { max_depth: None, min_samples_leaf: 1, min_samples_split: 2, - n_trees: 100, // this is n_estimators in sklearn + n_trees: 100, // this is n_estimators in sklearn m: Option::None, keep_samples: false, seed: 0,