fix: code cleanup
This commit is contained in:
@@ -144,7 +144,7 @@ mod tests {
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn my_fit_longley1() {
|
||||
fn my_fit_longley_ndarray() {
|
||||
|
||||
let x = arr2(&[
|
||||
[ 234.289, 235.6, 159., 107.608, 1947., 60.323],
|
||||
@@ -163,9 +163,7 @@ mod tests {
|
||||
[ 502.601, 393.1, 251.4, 125.368, 1960., 69.564],
|
||||
[ 518.173, 480.6, 257.2, 127.852, 1961., 69.331],
|
||||
[ 554.894, 400.7, 282.7, 130.081, 1962., 70.551]]);
|
||||
let y = arr1(&[83.0, 88.5, 88.2, 89.5, 96.2, 98.1, 99.0, 100.0, 101.2, 104.6, 108.4, 110.8, 112.6, 114.2, 115.7, 116.9]);
|
||||
|
||||
println!("{:?}", y.shape());
|
||||
let y = arr1(&[83.0, 88.5, 88.2, 89.5, 96.2, 98.1, 99.0, 100.0, 101.2, 104.6, 108.4, 110.8, 112.6, 114.2, 115.7, 116.9]);
|
||||
|
||||
let expected_y: Vec<f64> = vec![85., 88., 88., 89., 97., 98., 99., 99., 102., 104., 109., 110., 113., 114., 115., 116.];
|
||||
|
||||
@@ -174,9 +172,7 @@ mod tests {
|
||||
min_samples_leaf: 1,
|
||||
min_samples_split: 2,
|
||||
n_trees: 1000,
|
||||
mtry: Option::None}).predict(&x);
|
||||
|
||||
println!("{:?}", y_hat);
|
||||
mtry: Option::None}).predict(&x);
|
||||
|
||||
for i in 0..y_hat.len() {
|
||||
assert!((y_hat[i] - expected_y[i]).abs() < 1.0);
|
||||
|
||||
Reference in New Issue
Block a user