Moved activations module and update imports in dense_nn and logreg

This commit is contained in:
Palash Tyagi 2025-07-06 23:50:32 +01:00
parent 4f8a27298c
commit 87d14bbf5f
5 changed files with 8 additions and 6 deletions

View File

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

View File

@ -130,6 +130,6 @@ mod tests {
fn test_dleaky_relu_edge_case() { fn test_dleaky_relu_edge_case() {
let x = Matrix::from_vec(vec![-1e-10, 0.0, 1e10], 3, 1); 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); 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);
} }
} }

View File

@ -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 crate::matrix::{Matrix, SeriesOps};
use rand::prelude::*; use rand::prelude::*;

View File

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

View File

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