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

apply fmt

Browse Source
This commit is contained in:
Lorenzo (Mec-iS)
2022-10-31 19:28:24 +00:00
parent a9f89a2e15
commit 78e53a28e7
1 changed files with 2 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/ensemble/random_forest_classifier.rs
+2 -4
View File
@@ -820,9 +820,7 @@ mod tests {
&[6.6, 2.9, 4.6, 1.3], &[6.6, 2.9, 4.6, 1.3],
&[5.2, 2.7, 3.9, 1.4], &[5.2, 2.7, 3.9, 1.4],
]); ]);
let y = vec![ let y = vec![0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1];
0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
];
let classifier = RandomForestClassifier::fit( let classifier = RandomForestClassifier::fit(
&x, &x,
@@ -832,7 +830,7 @@ mod tests {
max_depth: None, max_depth: None,
min_samples_leaf: 1, min_samples_leaf: 1,
min_samples_split: 2, 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, m: Option::None,
keep_samples: false, keep_samples: false,
seed: 0, seed: 0,