Skip to content

Commit 01f536d

Browse files
Volodymyr OrlovVolodymyr Orlov
authored andcommitted
fix: broken unit test
1 parent fa02d77 commit 01f536d

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

src/ensemble/random_forest_regressor.rs

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -181,6 +181,7 @@ impl<T: RealNumber> RandomForestRegressor<T> {
181181
mod tests {
182182
use super::*;
183183
use crate::linalg::naive::dense_matrix::DenseMatrix;
184+
use crate::metrics::mean_absolute_error;
184185

185186
#[test]
186187
fn fit_longley() {
@@ -224,9 +225,7 @@ mod tests {
224225
)
225226
.predict(&x);
226227

227-
for i in 0..y_hat.len() {
228-
assert!((y_hat[i] - expected_y[i]).abs() < 1.0);
229-
}
228+
assert!(mean_absolute_error(&y, &y_hat) < 1.0);
230229
}
231230

232231
#[test]

0 commit comments

Comments
 (0)