Replace calls to .sum() with .sum_axis()

tspooner · Sep 12, 2017 · 615891e · 615891e
1 parent aa704e9
commit 615891e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/fa/projection/rbf_network.rs b/src/fa/projection/rbf_network.rs
@@ -50,8 +50,8 @@ impl RBFNetwork {
 impl Projection<RegularSpace<Continuous>> for RBFNetwork {
     fn project(&self, input: &Vec<f64>) -> Array1<f64> {
         let d = &self.mu - &ArrayView::from_shape((1, self.mu.cols()), input).unwrap();
-        let e = (&d * &d * &self.gamma).mapv(|v| v.exp()).sum(Axis(1));
-        let z = e.sum(Axis(0));
+        let e = (&d * &d * &self.gamma).mapv(|v| v.exp()).sum_axis(Axis(1));
+        let z = e.sum_axis(Axis(0));
 
         e / z
     }