lluni/rust-nn
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Optimized MSE

Browse source
This commit is contained in:
lluni 2023-01-15 23:54:16 +01:00
parent cd47bd7d12
commit 58964ceb41
Signed by: lluni
GPG key ID: ACEEB468BC325D35
1 changed files with 1 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
src/functions/loss_functions.rs
View file

@ -1,5 +1,3 @@
use std::ops::MulAssign;
use ndarray::{Array1, ArrayView1}; use ndarray::{Array1, ArrayView1};
pub enum Type { pub enum Type {
@ -16,7 +14,7 @@ pub fn parse_type(t: Type) -> (fn(ArrayView1<f64>, ArrayView1<f64>) -> f64, fn(A
pub fn mse(y_true: ArrayView1<f64>, y_pred: ArrayView1<f64>) -> f64 { pub fn mse(y_true: ArrayView1<f64>, y_pred: ArrayView1<f64>) -> f64 {
let mut temp = &y_true - &y_pred; let mut temp = &y_true - &y_pred;
temp.mul_assign(&temp.clone()); temp.mapv_inplace(|x| x * x);
let mut sum = 0.0; let mut sum = 0.0;
for i in 0..temp.len() { for i in 0..temp.len() {
sum += temp.get(i).unwrap(); sum += temp.get(i).unwrap();