Moved activations module and update imports in dense_nn and logreg

src/compute/mod.rs

@@ -1,3 +1,4 @@
-pub mod activations;
 
-pub mod models;
+pub mod models;
+
+pub mod stats;

src/compute/models/activations.rs

@@ -130,6 +130,6 @@ mod tests {
     fn test_dleaky_relu_edge_case() {
         let x = Matrix::from_vec(vec![-1e-10, 0.0, 1e10], 3, 1);
         let expected = Matrix::from_vec(vec![0.01, 0.01, 1.0], 3, 1);
-        assert_eq!(dleaky_relu(&x), expected);
+        assert_eq!(dleaky_relu(&x), expectewd);
     }
 }

src/compute/models/dense_nn.rs

@@ -1,4 +1,4 @@
-use crate::compute::activations::{drelu, relu, sigmoid};
+use crate::compute::models::activations::{drelu, relu, sigmoid};
 use crate::matrix::{Matrix, SeriesOps};
 use rand::prelude::*;
 

src/compute/models/logreg.rs

@@ -1,4 +1,4 @@
-use crate::compute::activations::sigmoid;
+use crate::compute::models::activations::sigmoid;
 use crate::matrix::{Matrix, SeriesOps};
 
 pub struct LogReg {

src/compute/models/mod.rs

@@ -3,4 +3,5 @@ pub mod logreg;
 pub mod dense_nn;
 pub mod k_means;    
 pub mod pca;
-pub mod gaussian_nb;
+pub mod gaussian_nb;
+pub mod activations;