fix: code cleanup

This commit is contained in:
Volodymyr Orlov
2020-03-26 16:23:50 -07:00
parent 4d967858a5
commit 1257d2c19b
2 changed files with 4 additions and 37 deletions
+3 -7
View File
@@ -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);