feat: add basic Matrix implementation for ndarray

2019-12-23 10:33:19 -08:00
parent 2425419d10
commit c1d7c038a6
6 changed files with 545 additions and 15 deletions
@@ -27,7 +27,7 @@ impl<M: Matrix> LinearRegression<M> {
        }
        
        let b = y.clone();
-        let mut a = x.h_stack(&M::ones(x_nrows, 1));
+        let mut a = x.v_stack(&M::ones(x_nrows, 1));

        let w = match solver {
            LinearRegressionSolver::QR => a.qr_solve_mut(b),