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 {
use super::*;
use crate::linalg::naive::dense_matrix::DenseMatrix;
use crate::metrics::*;
#[test]
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]
+1 -1
View File
@@ -735,7 +735,7 @@ mod tests {
min_samples_leaf: 1,
min_samples_split: 2,
n_trees: 1000,
mtry: Option::None,
m: Option::None,
},
)
.predict(&x);