magnus167/rustframe
1
0
Fork 0
mirror of https://github.com/Magnus167/rustframe.git synced 2025-10-04 13:19:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

applied formatting

Browse Source
This commit is contained in:
Palash Tyagi
2025-07-06 20:43:01 +01:00
parent 75d07371b2
commit 46abeb12a7
1 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/compute/models/dense_nn.rs
View File

@@ -1,5 +1,5 @@
use crate::compute::activations::{drelu, relu, sigmoid};
use crate::matrix::{Matrix, SeriesOps};
use crate::compute::activations::{relu, drelu, sigmoid};
use rand::prelude::*;
/// Supported activation functions
@@ -167,7 +167,11 @@ impl DenseNN {
LossKind::BCE => self.loss.gradient(&y_hat, y),
LossKind::MSE => {
let grad = self.loss.gradient(&y_hat, y);
let dz = self.activations.last().unwrap().derivative(zs.last().unwrap());
let dz = self
.activations
.last()
.unwrap()
.derivative(zs.last().unwrap());
grad.zip(&dz, |g, da| g * da)
}
};