Fixed test by transposing matrix

smartcorelib · Aug 29, 2022 · 7f7b2ed · 7f7b2ed
1 parent d46b830
commit 7f7b2ed
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/ensemble/random_forest_classifier.rs b/src/ensemble/random_forest_classifier.rs
@@ -570,15 +570,15 @@ mod tests_prob {
         assert_eq!(
             results,
             DenseMatrix::<f64>::from_array(
-                20,
                 2,
+                20,
                 &[
                     1.0, 0.78, 0.95, 0.82, 1.0, 0.92, 0.99, 0.96, 0.36, 0.33, 0.02, 0.02, 0.0, 0.0,
                     0.0, 0.0, 0.03, 0.05, 0.0, 0.02, 0.0, 0.22, 0.05, 0.18, 0.0, 0.08, 0.01, 0.04,
                     0.64, 0.67, 0.98, 0.98, 1.0, 1.0, 1.0, 1.0, 0.97, 0.95, 1.0, 0.98
                 ]
             )
+            .transpose()
         );
-        assert!(false);
     }
 }