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

fix: minor refactoring

Browse Source
This commit is contained in:
Volodymyr Orlov
2020-03-23 18:50:58 -07:00
parent 17200fe633
commit 18243e658b
2 changed files with 14 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/ensemble/random_forest_classifier.rs
+2 -2
View File
@@ -9,7 +9,7 @@ use crate::tree::decision_tree_classifier::{DecisionTreeClassifier, DecisionTree
pub struct RandomForestClassifierParameters {
pub criterion: SplitCriterion,
pub max_depth: Option<u16>,
pub min_samples_leaf: u16,
pub min_samples_leaf: usize,
pub min_samples_split: usize,
pub n_trees: u16,
pub mtry: Option<usize>
@@ -97,7 +97,7 @@ impl RandomForestClassifier {
}
fn sample_with_replacement(y: &Vec<usize>, num_classes: usize) -> Vec<u32>{
fn sample_with_replacement(y: &Vec<usize>, num_classes: usize) -> Vec<usize>{
let mut rng = rand::thread_rng();
let class_weight = vec![1.; num_classes];
let nrows = y.len();