fix: broken unit tests

This commit is contained in:
Volodymyr Orlov
2020-09-04 10:10:41 -07:00
parent ecfbaac167
commit 1735ea61df
2 changed files with 4 additions and 2 deletions
+3 -1
View File
@@ -222,6 +222,7 @@ impl<T: RealNumber> RandomForestClassifier<T> {
mod tests { mod tests {
use super::*; use super::*;
use crate::linalg::naive::dense_matrix::DenseMatrix; use crate::linalg::naive::dense_matrix::DenseMatrix;
use crate::metrics::*;
#[test] #[test]
fn fit_predict_iris() { fn fit_predict_iris() {
@@ -264,7 +265,8 @@ mod tests {
}, },
); );
assert_eq!(y, classifier.predict(&x)); assert!(accuracy(&y, &classifier.predict(&x)) > 0.9);
} }
#[test] #[test]
+1 -1
View File
@@ -735,7 +735,7 @@ mod tests {
min_samples_leaf: 1, min_samples_leaf: 1,
min_samples_split: 2, min_samples_split: 2,
n_trees: 1000, n_trees: 1000,
mtry: Option::None, m: Option::None,
}, },
) )
.predict(&x); .predict(&x);