We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent fa02d77 commit 01f536dCopy full SHA for 01f536d
src/ensemble/random_forest_regressor.rs
@@ -181,6 +181,7 @@ impl<T: RealNumber> RandomForestRegressor<T> {
181
mod tests {
182
use super::*;
183
use crate::linalg::naive::dense_matrix::DenseMatrix;
184
+ use crate::metrics::mean_absolute_error;
185
186
#[test]
187
fn fit_longley() {
@@ -224,9 +225,7 @@ mod tests {
224
225
)
226
.predict(&x);
227
- for i in 0..y_hat.len() {
228
- assert!((y_hat[i] - expected_y[i]).abs() < 1.0);
229
- }
+ assert!(mean_absolute_error(&y, &y_hat) < 1.0);
230
}
231
232
0 commit comments