fix: broken unit test
This commit is contained in:
@@ -181,6 +181,7 @@ impl<T: RealNumber> RandomForestRegressor<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::mean_absolute_error;
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn fit_longley() {
|
fn fit_longley() {
|
||||||
@@ -224,9 +225,7 @@ mod tests {
|
|||||||
)
|
)
|
||||||
.predict(&x);
|
.predict(&x);
|
||||||
|
|
||||||
for i in 0..y_hat.len() {
|
assert!(mean_absolute_error(&y, &y_hat) < 1.0);
|
||||||
assert!((y_hat[i] - expected_y[i]).abs() < 1.0);
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
|
|||||||
Reference in New Issue
Block a user