naive_mean_absolute_error

carrotflakes · Sep 20, 2023 · 5663d2a · 5663d2a
1 parent 01ad9cb
commit 5663d2a
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/losses.rs b/src/losses.rs
@@ -12,6 +12,12 @@ pub fn naive_mean_squared_error(x0: ComputedNDA, x1: ComputedNDA) -> ComputedNDA
         / ComputedNDA::new(scalar(x.shape().iter().product::<usize>() as f32))
 }
 
+pub fn naive_mean_absolute_error(x0: ComputedNDA, x1: ComputedNDA) -> ComputedNDA {
+    let x = (x0 - x1).abs();
+    x.sum(Vec::from_iter(0..x.ndim()), false)
+        / ComputedNDA::new(scalar(x.shape().iter().product::<usize>() as f32))
+}
+
 pub fn softmax_cross_entropy(t: Vec<usize>, x: &ComputedNDA) -> ComputedNDA {
     let n = x.shape().iter().take(x.ndim() - 1).product();
     let log_z = log_sum_exp(&*x);